minor...

modules/deeplearning/icing_fcn.py

@@ -736,6 +736,7 @@ class IcingIntensityFCN:
         inputs = [mini_batch[0], mini_batch[1]]
         labels = mini_batch[2]
         pred = self.model(inputs, training=False)
+        pred = tf.reshape(pred, (pred.shape[0], NumLogits))
         t_loss = self.loss(labels, pred)
 
         self.test_labels.append(labels)