Skip to content

Instantly share code, notes, and snippets.

@vsay01

vsay01/checkpoint.py

Created January 22, 2020 23:23
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 vsay01/ea9d8da0af543eb64918c1d4f83624df to your computer and use it in GitHub Desktop.
Save vsay01/ea9d8da0af543eb64918c1d4f83624df to your computer and use it in GitHub Desktop.
Download ZIP
checkpoint data
Raw
checkpoint.py
checkpoint = {
'epoch': epoch + 1,
'valid_loss_min': valid_loss,
'state_dict': model.state_dict(),
'optimizer': optimizer.state_dict(),
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment