Skip to content

Instantly share code, notes, and snippets.

@rohan-varma

rohan-varma/named_modules.py

Created February 25, 2021 22:37
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save rohan-varma/b8e403dc6b1b0f058d36f966b4ca3778 to your computer and use it in GitHub Desktop.
Save rohan-varma/b8e403dc6b1b0f058d36f966b4ca3778 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
named_modules.py
for mn, m in net.named_modules():
for param_name, param in m.named_parameters(recurse=False):
qualname = f"{mn}.{param_name}"
print(f"Got qualname {qualname}")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment