Merge pull request #1 from DIAGNijmegen/MatinHz-patch-1

Update tfdata.shuffle buffer_size
DIAGNijmegen · Nov 4, 2021 · d6eaf0e · d6eaf0e
2 parents 999ee75 + bf60fab
commit d6eaf0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tf2.5/scripts/train_model.py b/tf2.5/scripts/train_model.py
@@ -175,7 +175,7 @@
                                                                           output_shapes = EXPECTED_IO_SHAPE)       # Initialize TensorFlow Dataset
     if str(args.CACHE_TDS_PATH)!='None': 
         train_gen = train_gen.cache(filename=(None if str(args.CACHE_TDS_PATH)=='None' else args.CACHE_TDS_PATH))  # Cache Dataset on Remote Server
-    train_gen     = train_gen.shuffle(int(TRAIN_DATA_SAMPLES*0.50))                                                # Shuffle Samples
+    train_gen     = train_gen.shuffle(int(TRAIN_DATA_SAMPLES))                                                     # Shuffle Samples
     train_gen     = train_gen.map(lambda x,y: augment_tensors(x,y,args.AUGM_PARAMS,args.TRAIN_OBJ), 
                                                               num_parallel_calls=multiprocessing.cpu_count())
     train_gen     = train_gen.batch(args.BATCH_SIZE)                                                               # Load Data in Batches