n0obcoder/understanding_pytorch_12_load_state_dict.py

Last active August 12, 2019 18:15

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

Download ZIP

Raw

understanding_pytorch___12___load_state_dict.py

	model_new = NeuralNet()
	model_new.load_state_dict(torch.load('weights_only.pth'))

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