diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index ac3bb5ef31fee3e834c5d9aaa734f1e8d0ba618b..574ce669471293e982e9601053590c6db5380d1a 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/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)