khalidmeister/save.model.py

Created June 23, 2020 14:55

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

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

Download ZIP

Save the model

Raw

save.model.py

	# Save The Model
	PATH = './fix_alexnet_net.pth'
	torch.save(model_alexnet.state_dict(), PATH)

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