Skip to content

Commit daacd1d

Browse files
Ryan Sepassicopybara-github
authored andcommitted
Change readme example to explicitly set shuffle_files
PiperOrigin-RevId: 279801514
1 parent 1b5735d commit daacd1d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ tf.compat.v1.enable_eager_execution()
5353
print(tfds.list_builders())
5454

5555
# Construct a tf.data.Dataset
56-
ds_train, ds_test = tfds.load(name="mnist", split=["train", "test"])
56+
ds_train = tfds.load(name="mnist", split="train", shuffle_files=True)
5757

5858
# Build your input pipeline
5959
ds_train = ds_train.shuffle(1000).batch(128).prefetch(10)

0 commit comments

Comments
 (0)