snapshot...

modules/deeplearning/srcnn_l1b_l2.py

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