Skip to content

Instantly share code, notes, and snippets.

@jamescalam

jamescalam/meditations_save_model.py

Created April 1, 2020 16:40
Show Gist options
  • Save jamescalam/503f1fbf38a48c1b109e47e1af538535 to your computer and use it in GitHub Desktop.
Save jamescalam/503f1fbf38a48c1b109e47e1af538535 to your computer and use it in GitHub Desktop.
Download ZIP
Snippet showing how to setup model checkpoint saving using built-in call backs function
Raw
meditations_save_model.py
checkpoint = tf.keras.callbacks.ModelCheckpoint(
filepath='./training_checkpoints/ckpt_{epoch}',
save_weights_only=True
)
history = model.fit(dataset, epochs=EPOCHS, callbacks=[checkpoint_callback])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment