diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 574ce669471293e982e9601053590c6db5380d1a..0e4b864c9c596ac34b47dea591b81ffcc36b5f29 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -514,9 +514,11 @@ 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)
+        in_nd = mini_batch.numpy()
+        print('****: ', in_nd.shape, in_nd.min(), in_nd.max())
         pred = self.model(inputs, training=False)
-        print('****: ', pred.shape, pred.numpy().min(), pred.numpy().max())
+        in_nd = pred.numpy()
+        print('****: ', in_nd.shape, in_nd.min(), in_nd.max())
         t_loss = self.loss(labels, pred)
 
         self.test_loss(t_loss)