From f78fa93f2c374f03c282590c672e0465e927aebf Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 26 Dec 2022 14:07:42 -0600 Subject: [PATCH] snapshot... --- modules/deeplearning/srcnn_l1b_l2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index ac3bb5ef..574ce669 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -514,6 +514,7 @@ 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) pred = self.model(inputs, training=False) print('****: ', pred.shape, pred.numpy().min(), pred.numpy().max()) t_loss = self.loss(labels, pred) -- GitLab