Code: Select all
ds = tf.data.Dataset.from_generator(my_generator)
ds = ds.prefetch(tf.data.AUTOTUNE).shuffle(1000).batch(128).cache()
Code: Select all
ds = tf.data.Dataset.from_generator(my_generator)
ds = ds.prefetch(tf.data.AUTOTUNE).shuffle(1000).batch(128).cache()