snapshot...

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