Skip to content

Instantly share code, notes, and snippets.

@lxuechen
Created August 3, 2018 19:41
Show Gist options
  • Save lxuechen/cb8dc59b8b88877701ae0210460dd9a9 to your computer and use it in GitHub Desktop.
Save lxuechen/cb8dc59b8b88877701ae0210460dd9a9 to your computer and use it in GitHub Desktop.
simple looping over dataset in eager
for image, label in dataset:
logits = model(image, training=True)
...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment