khalidmeister/load-model.py

Created June 23, 2020 14:57

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/72c316c01153a91dbb1293dd5b26ed06.js"></script>
Save khalidmeister/72c316c01153a91dbb1293dd5b26ed06 to your computer and use it in GitHub Desktop.

Download ZIP

Load the model

Raw

load-model.py

	# Declare the network
	net = Net()

	# Set the parameters
	net.load_state_dict(torch.load(PATH))

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