Skip to content

Instantly share code, notes, and snippets.

@sayakpaul

sayakpaul/bs_tuning.py

Created April 29, 2020 04:02
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save sayakpaul/119c9f3b30f8bbb3b19242448efaa8b9 to your computer and use it in GitHub Desktop.
Save sayakpaul/119c9f3b30f8bbb3b19242448efaa8b9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
bs_tuning.py
# Calculate batch size
batch_size_per_replica = 32
batch_size = batch_size_per_replica * strategy.num_replicas_in_sync
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment