Skip to content

Instantly share code, notes, and snippets.

@MohanaRC
Created June 5, 2023 14:02
Show Gist options
  • Save MohanaRC/61c2b6bdc94fa71cd669a9757cc3db25 to your computer and use it in GitHub Desktop.
Save MohanaRC/61c2b6bdc94fa71cd669a9757cc3db25 to your computer and use it in GitHub Desktop.
batch_size = 64
train_data = train_data.map(format_image)
test_data = test_data.map(format_image)
train = train_data.shuffle(buffer_size=1024).batch(batch_size)
test = test_data.batch(batch_size=batch_size)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment