n0obcoder/understanding_pytorch_14_save_entire_model.py

Last active August 12, 2019 18:14

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/n0obcoder/f585f87eb78a7db208b9218d7bc8e706.js"></script>
Save n0obcoder/f585f87eb78a7db208b9218d7bc8e706 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

understanding_pytorch___14___save_entire_model.py

torch.save(model, 'entire_model.pth')

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment