From 390a0b44ae892a4c291f6dd744b4c98abf3faa41 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 12 Dec 2022 11:40:12 -0600
Subject: [PATCH] snapshot..
---
modules/deeplearning/srcnn_l1b_l2.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index f97a4865..64701b47 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -664,7 +664,7 @@ class SRCNN:
labels_denorm = denormalize(labels, label_param, mean_std_dct)
preds_denorm = denormalize(preds, label_param, mean_std_dct)
- # return labels_denorm, preds_denorm
+ return labels_denorm, preds_denorm
def do_evaluate(self, data, ckpt_dir):
@@ -700,7 +700,7 @@ class SRCNN:
self.build_model()
self.build_training()
self.build_evaluation()
- self.restore(ckpt_dir)
+ return self.restore(ckpt_dir)
def run_evaluate(self, data, ckpt_dir):
self.num_data_samples = 80000
@@ -712,7 +712,7 @@ class SRCNN:
def run_restore_static(directory, ckpt_dir):
nn = SRCNN()
- nn.run_restore(directory, ckpt_dir)
+ return nn.run_restore(directory, ckpt_dir)
def run_evaluate_static(in_file, out_file, ckpt_dir):
--
GitLab