From 40700f3544efdbc0598739cbf8986e81d08a5972 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 4 Feb 2024 11:10:51 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index eac05cb0..c07277d1 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -2,6 +2,7 @@ import tensorflow as tf
from util.setup import logdir, modeldir, cachepath, now, ancillary_path, home_dir
from util.util import EarlyStop, normalize
from icing.util import make_for_full_domain_predict
+from util.augment import augment_icing
from util.geos_nav import get_navigation, get_lon_lat_2d_mesh
import os, datetime
@@ -440,8 +441,10 @@ 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, reshuffle_each_iteration=True)
if DO_AUGMENT:
dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
+ # dataset = dataset.map(augment_icing(), num_parallel_calls=8)
dataset = dataset.prefetch(buffer_size=1)
self.train_dataset = dataset
--
GitLab