sleepingcat4/param_counter.py

Created November 24, 2023 12:56

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

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

Download ZIP

we're counting params using torchscale

Raw

param_counter.py

	total_params = sum(p.numel() for p in model.parameters())
	print(f"Total parameters in the model: {total_params}")

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