From 6fa5af8fa562aa649a5cad3b0b0cc647085ca3a4 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 26 Dec 2022 14:45:52 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn_l1b_l2.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 0e4b864c..49c0d7ad 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/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)
--
GitLab