jojonki/print_training_params.py

Last active December 7, 2017 15:06

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

Download ZIP

Print training parameters in PyTorch

Raw

print_training_params.py

	for name, param in model.named_parameters():
	if param.requires_grad:
	print(name, param.data)

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