Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created December 24, 2020 15:13
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 amankharwal/e95f1ca280e30f268faa4a1013650f06 to your computer and use it in GitHub Desktop.
Save amankharwal/e95f1ca280e30f268faa4a1013650f06 to your computer and use it in GitHub Desktop.
STEP_SIZE_TRAIN = int(np.ceil(train_generator.n / train_generator.batch_size))
STEP_SIZE_VAL = int(np.ceil(validation_generator.n / validation_generator.batch_size))
print("Train step size:", STEP_SIZE_TRAIN)
print("Validation step size:", STEP_SIZE_VAL)
train_generator.reset()
validation_generator.reset()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment