Exemple #1
0
 def dataset_fn(ds):
     if id == 0:
         ds = ds.batch(4)
     elif id == 1:
         ds = ds.repeat(2)
     elif id == 2:
         ds = ds.batch(4)
         ds = ds.repeat(2)
     else:
         ds = ds.shuffle(buffer_size=4)
     return ds
Exemple #2
0
 def dataset_fn(ds):
     ds = ds.repeat(2)
     return ds.batch(4)