Skip to content

Instantly share code, notes, and snippets.

@sayakpaul

sayakpaul/checkpoints.py

Created April 29, 2020 11:45
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 sayakpaul/3f924a9afa3f89b0edc596cca6e7ebcb to your computer and use it in GitHub Desktop.
Save sayakpaul/3f924a9afa3f89b0edc596cca6e7ebcb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
checkpoints.py
# Set up model checkpoint callback
filepath = wandb.run.dir + "/{epoch:02d}-{val_accuracy:.2f}.ckpt"
checkpoint = tf.keras.callbacks.ModelCheckpoint(filepath,
monitor="val_accuracy",
verbose=1,
save_best_only=True, mode="max")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment