diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index b1daaa11d3ce9c789cc26ba1fb3d313e1e06e87b..e59913c045a19ed0dc0dbbc361a8ebd202a911fd 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -442,7 +442,7 @@ class SRCNN:
         dataset = dataset.batch(PROC_BATCH_SIZE)
         dataset = dataset.map(self.data_function, num_parallel_calls=8)
         dataset = dataset.cache(filename=CACHE_FILE)
-        dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE, reshuffle_each_iteration=True)
+        dataset = dataset.shuffle(1, reshuffle_each_iteration=True)
         if DO_AUGMENT:
             dataset = dataset.map(augment_image(), num_parallel_calls=8)
         dataset = dataset.prefetch(buffer_size=1)
diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index 63a0ef18722e0857360d2ad2463a9248a56f241e..648c47cca75aeaa2cac2046f2db5d07d686b83f5 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -467,7 +467,7 @@ class SRCNN:
         dataset = dataset.batch(PROC_BATCH_SIZE)
         dataset = dataset.map(self.data_function, num_parallel_calls=8)
         dataset = dataset.cache(filename=CACHE_FILE)
-        dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE, reshuffle_each_iteration=True)
+        dataset = dataset.shuffle(1, reshuffle_each_iteration=True)
         if DO_AUGMENT:
             dataset = dataset.map(augment_image(), num_parallel_calls=8)
         dataset = dataset.prefetch(buffer_size=1)