Skip to content
Snippets Groups Projects
Commit 987d80eb authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 69f178e6
Branches
No related tags found
No related merge requests found
......@@ -435,7 +435,8 @@ 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)
if DO_AUGMENT:
dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
dataset = dataset.prefetch(buffer_size=1)
self.train_dataset = dataset
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment