Created
April 29, 2020 04:01
-
-
Save sayakpaul/63a085101f3691049dff677bae7908f6 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Prepare batches and randomly shuffle the training images (this time with prefetch) | |
train_batches = train.shuffle(1024).repeat().batch(batch_size).\ | |
prefetch(tf.data.experimental.AUTOTUNE) | |
valid_batches = valid.repeat().batch(batch_size).\ | |
prefetch(tf.data.experimental.AUTOTUNE) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment