From 7d48a97e3f3adb116d106ac09fc0e0a5ab005adc Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 6 Jun 2022 08:54:33 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/unet_l1b_l2.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index e94fd5ff..fc418e80 100644
--- a/modules/deeplearning/unet_l1b_l2.py
+++ b/modules/deeplearning/unet_l1b_l2.py
@@ -370,7 +370,7 @@ class UNET:
# print('num test samples: ', tst_idxs.shape[0])
# print('setup_pipeline: Done')
- def setup_pipeline(self, train_data_files, train_label_files, test_data_files, test_label_files):
+ def setup_pipeline(self, train_data_files, train_label_files, test_data_files, test_label_files, num_train_samples):
self.train_data_files = train_data_files
self.train_label_files = train_label_files
@@ -384,7 +384,7 @@ class UNET:
self.get_train_dataset(trn_idxs)
self.get_test_dataset(tst_idxs)
- self.num_data_samples = 34000 # approximately
+ self.num_data_samples = num_train_samples # approximately
print('datetime: ', now)
print('training and test data: ')
--
GitLab