Skip to content

Instantly share code, notes, and snippets.

@vsay01

vsay01/continue_training.py

Created January 22, 2020 13:03
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/890b2149bc48a1cf983fd93b8baea8ac to your computer and use it in GitHub Desktop.
Save vsay01/890b2149bc48a1cf983fd93b8baea8ac to your computer and use it in GitHub Desktop.
Download ZIP
Continue training
Raw
continue_training.py
trained_model = train(start_epoch, 6, valid_loss_min, loaders, model, optimizer, criterion, use_cuda, "./checkpoint/current_checkpoint.pt", "./best_model/best_model.pt")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment