diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 02695b5bce1913deceb1306d005101821ddc1b3a..dc61cbc8791424044a9887391dd017bce70265f2 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -12,6 +12,7 @@ import h5py LOG_DEVICE_PLACEMENT = False +EVAL_BATCH_SIZE = 8192 PROC_BATCH_SIZE = 4096 PROC_BATCH_BUFFER_SIZE = 50000 @@ -489,7 +490,7 @@ class IcingIntensityFCN: indexes = list(indexes) dataset = tf.data.Dataset.from_tensor_slices(indexes) - dataset = dataset.batch(PROC_BATCH_SIZE) + dataset = dataset.batch(EVAL_BATCH_SIZE) dataset = dataset.map(self.data_function_evaluate, num_parallel_calls=8) dataset = dataset.cache() self.eval_dataset = dataset