From 31a4ecb1176221bbb86f3b66183f76fa88fbeb1d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 6 Mar 2023 12:18:38 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn_l1b_l2.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 10d32ca2..02c30c6b 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -518,7 +518,7 @@ class SRCNN:
def predict(self, mini_batch):
inputs = [mini_batch[0]]
labels = mini_batch[1]
- pred = self.model(inputs, training=False)
+ pred = self.model([inputs], training=False)
t_loss = self.loss(labels, pred)
self.test_labels.append(labels)
--
GitLab