czs0x55aa/print_model_parameters.py

Created September 12, 2017 14:08

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

Download ZIP

print model parameters in pytorch

Raw

print_model_parameters.py

	for name, param in model.state_dict().items():
	print(name, param.size())

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