Skip to content

Instantly share code, notes, and snippets.

@vsay01

vsay01/load_saved_checkpoint_param.py

Last active January 22, 2020 23:29
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/ed3812306c00f129e5d738686cdefd7b to your computer and use it in GitHub Desktop.
Save vsay01/ed3812306c00f129e5d738686cdefd7b to your computer and use it in GitHub Desktop.
Download ZIP
Load saved checkpoint param
Raw
load_saved_checkpoint_param.py
# define optimzer
optimizer = optim.Adam(model.parameters(), lr=0.001)
# define checkpoint saved path
ckp_path = "./checkpoint/current_checkpoint.pt"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment