From 3d7b97ae906f650adf18a853b485aa2e4df8dd77 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 12 Dec 2022 11:27:35 -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 2b075def..f97a4865 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -694,7 +694,7 @@ class SRCNN:
self.do_training(ckpt_dir=ckpt_dir)
def run_restore(self, directory, ckpt_dir):
- valid_data_files = glob.glob(directory + 'data_*.npy')
+ valid_data_files = glob.glob(directory + 'data_valid*.npy')
self.num_data_samples = 1000
self.setup_test_pipeline(valid_data_files)
self.build_model()
--
GitLab