import tensorflow as tf

            aux_logits = self._add_fully_connected(X, (ch,), K, no_reg=True)

        return aux_logits

    def _compute_predictions(self, logits, classes):
        probs = tf.nn.softmax(logits)
        preds = tf.argmax(logits, axis=1, output_type=tf.int32)
        corrects = tf.equal(preds, classes)
        return (probs, corrects)

    def _compute_loss(self, logits, aux_logits_list, classes, **knobs):
        reg_decay = knobs['reg_decay']
        aux_loss_mul = knobs['aux_loss_mul'] # Multiplier for auxiliary loss

