Skip to content

Instantly share code, notes, and snippets.

@1pha

1pha/count_params.py

Created October 19, 2022 01:02
Show Gist options
  • Save 1pha/577206868960cb4ca3146e737dd7a319 to your computer and use it in GitHub Desktop.
Save 1pha/577206868960cb4ca3146e737dd7a319 to your computer and use it in GitHub Desktop.
Download ZIP
Count Pytorch Model Parameters
Raw
count_params.py
# Reference
# https://discuss.pytorch.org/t/how-do-i-check-the-number-of-parameters-of-a-model/4325/8
def count_parameters(model):
return sum(p.numel() for p in model.parameters() if p.requires_grad)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment