From 6dfa5a30e7856b4a1320d6328138a935142fd6ea Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 7 Oct 2022 10:23:49 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn_l1b_l2.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 5e61e92b..136de6fe 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -641,7 +641,7 @@ class SRCNN:
return pred
- def run(self, directory):
+ def run(self, directory, ckpt_dir=None):
train_data_files = glob.glob(directory+'data_train*.npy')
valid_data_files = glob.glob(directory+'data_valid*.npy')
@@ -649,7 +649,7 @@ class SRCNN:
self.build_model()
self.build_training()
self.build_evaluation()
- self.do_training()
+ self.do_training(ckpt_dir=ckpt_dir)
def run_restore(self, directory, ckpt_dir):
valid_data_files = glob.glob(directory + 'data_*.npy')
--
GitLab