Skip to content
Snippets Groups Projects
Commit 40700f35 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent d942b75b
No related branches found
No related tags found
No related merge requests found
......@@ -2,6 +2,7 @@ import tensorflow as tf
from util.setup import logdir, modeldir, cachepath, now, ancillary_path, home_dir
from util.util import EarlyStop, normalize
from icing.util import make_for_full_domain_predict
from util.augment import augment_icing
from util.geos_nav import get_navigation, get_lon_lat_2d_mesh
import os, datetime
......@@ -440,8 +441,10 @@ 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, reshuffle_each_iteration=True)
if DO_AUGMENT:
dataset = dataset.shuffle(PROC_BATCH_BUFFER_SIZE)
# dataset = dataset.map(augment_icing(), num_parallel_calls=8)
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.
Finish editing this message first!
Please register or to comment