From 08ca488f86d895bfd6369f3f4a14f868eeb55066 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 3 Aug 2022 23:02:52 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index be6d471e..2b972f45 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -13,7 +13,7 @@ import h5py
LOG_DEVICE_PLACEMENT = False
-PROC_BATCH_SIZE = 10
+PROC_BATCH_SIZE = 4
PROC_BATCH_BUFFER_SIZE = 50000
NumClasses = 2
@@ -801,7 +801,7 @@ class ESPCN:
train_data_files.sort()
valid_data_files.sort()
- self.setup_pipeline(train_data_files, valid_data_files, 200000)
+ self.setup_pipeline(train_data_files, valid_data_files, 50000)
self.build_model()
self.build_training()
self.build_evaluation()
--
GitLab