From 6a5fcb9e5da2d50259a32460e373e4e7a1f7ab5b Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 7 Oct 2022 16:18:22 -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 7716228e..24a6fc44 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -638,11 +638,11 @@ class SRCNN:
return pred
- def run(self, directory, ckpt_dir=None):
+ def run(self, directory, ckpt_dir=None, num_data_samples=50000):
train_data_files = glob.glob(directory+'data_train*.npy')
valid_data_files = glob.glob(directory+'data_valid*.npy')
- self.setup_pipeline(train_data_files, valid_data_files, 50000)
+ self.setup_pipeline(train_data_files, valid_data_files, num_data_samples)
self.build_model()
self.build_training()
self.build_evaluation()
--
GitLab