jaypatravali/model_params.py

Created June 9, 2017 09:13

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

Download ZIP

print no of params in model

Raw

model_params.py

	# get the number of model parameters
	print('Number of model parameters: {}'.format(
	sum([p.data.nelement() for p in model.parameters()])))

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