Skip to content

Instantly share code, notes, and snippets.

@Gholamrezadar

Gholamrezadar/ghd-pytorch-snippets.py

Created May 26, 2024 18:09
Show Gist options
  • Save Gholamrezadar/b810ba72733c9ddea7cf075e53cadaa7 to your computer and use it in GitHub Desktop.
Save Gholamrezadar/b810ba72733c9ddea7cf075e53cadaa7 to your computer and use it in GitHub Desktop.
Download ZIP
PyTorch Snippets
Raw
ghd-pytorch-snippets.py
# get the number of parameters of the model
def get_num_params(model):
return sum(p.numel() for p in model.parameters())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment