snapshot...

modules/deeplearning/srcnn_l1b_l2.py

@@ -514,6 +514,7 @@ class SRCNN:
     def test_step(self, mini_batch):
         inputs = [mini_batch[0]]
         labels = mini_batch[1]
+        print('****: ', mini_batch[0].shape, mini_batch[0].min(), mini_batch[0].shape)
         pred = self.model(inputs, training=False)
         print('****: ', pred.shape, pred.numpy().min(), pred.numpy().max())
         t_loss = self.loss(labels, pred)