diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 4b380d85afc73331637a2a75f7d738a30fa77b60..ac3bb5ef31fee3e834c5d9aaa734f1e8d0ba618b 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -515,6 +515,7 @@ class SRCNN: inputs = [mini_batch[0]] labels = mini_batch[1] pred = self.model(inputs, training=False) + print('****: ', pred.shape, pred.numpy().min(), pred.numpy().max()) t_loss = self.loss(labels, pred) self.test_loss(t_loss)