Skip to content

Instantly share code, notes, and snippets.

@obeshor

obeshor/training_model.py

Created June 25, 2019 22:41
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 obeshor/0b5cbba3a5f4db2eaad71be09f31965c to your computer and use it in GitHub Desktop.
Save obeshor/0b5cbba3a5f4db2eaad71be09f31965c to your computer and use it in GitHub Desktop.
Download ZIP
Training Model
Raw
training_model.py
EPOCHS=10 #@param {type:"integer"}
STEPS_EPOCHS = train_generator.samples//train_generator.batch_size
VALID_STEPS=validation_generator.samples//validation_generator.batch_size
history = model.fit_generator(
train_generator,
steps_per_epoch=STEPS_EPOCHS,
epochs=EPOCHS,
validation_data=validation_generator,
validation_steps=VALID_STEPS)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment