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