From 4e8342231763b2fe8c6f12d4d30b227cf07d0df9 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 21 Mar 2022 15:33:27 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index a3159b64..781561e1 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -427,7 +427,8 @@ class IcingIntensityFCN:
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