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
def dataset_fn(ds):
ds = ds.map(operations = decode_op)
ds = ds.map(operations = normalize_op)
ds = ds.repeat(2)
return ds
def dataset_fn(ds):
ds = ds.repeat(2)
return ds.batch(4)