Skip to content

Instantly share code, notes, and snippets.

@kevakba
Created September 1, 2020 13:25
Show Gist options
  • Save kevakba/0a9f377d6a7c19e9310cc157cbf4dd87 to your computer and use it in GitHub Desktop.
Save kevakba/0a9f377d6a7c19e9310cc157cbf4dd87 to your computer and use it in GitHub Desktop.
epochs = 20
history = model.fit_generator(
train_data_gen,
steps_per_epoch=int(np.ceil(train_data_gen.n / float(batch_size))),
epochs=epochs,
validation_data=val_data_gen,
validation_steps=int(np.ceil(val_data_gen.n / float(batch_size)))
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment