n0obcoder/understanding_pytorch_15_load_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/f539676873bab027449367b559edb8a1.js"></script>
Save n0obcoder/f539676873bab027449367b559edb8a1 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

understanding_pytorch___15___load_entire_model.py

model_new = torch.load('entire_model.pth')

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