snapshot...

modules/deeplearning/srcnn_l1b_l2.py

@@ -518,7 +518,7 @@ class SRCNN:
     def predict(self, mini_batch):
         inputs = [mini_batch[0]]
         labels = mini_batch[1]
-        pred = self.model(inputs, training=False)
+        pred = self.model([inputs], training=False)
         t_loss = self.loss(labels, pred)
 
         self.test_labels.append(labels)