From 987d80eb90262613598c29a574ff8a75d0f9de68 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 23 Mar 2022 10:09:11 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index c05cdcff..e9175adc 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -435,7 +435,8 @@ class IcingIntensityNN:
dataset = dataset.batch(PROC_BATCH_SIZE)
dataset = dataset.map(self.data_function, num_parallel_calls=8)
dataset = dataset.cache()
- # dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
+ if DO_AUGMENT:
+ dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
dataset = dataset.prefetch(buffer_size=1)
self.train_dataset = dataset
--
GitLab