Skip to content

Instantly share code, notes, and snippets.

@FBosler

FBosler/model_training.py

Last active September 20, 2022 14:43
Show Gist options
  • Save FBosler/246846a39f6583eb1f717d46b03e7db0 to your computer and use it in GitHub Desktop.
Save FBosler/246846a39f6583eb1f717d46b03e7db0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_training.py
from keras_tqdm import TQDMNotebookCallback
batch_size = 32
history = model.fit_generator(
generator=training,
steps_per_epoch=training.samples // batch_size,
epochs=10,
callbacks=[TQDMNotebookCallback(leave_inner=True, leave_outer=True)],
validation_data=validation,
validation_steps=validation.samples // batch_size
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment