tutorials
tutorials copied to clipboard
NEW: beginner recipe on layer parameters
Fixes #2924 Re-implements #2914 (accidental deletion from forked repo)
Description
New beginner recipe on how to access model parameters, collect trainable parameters, and use torchinfo.summary()
- How to inspect a model's parameters using
.parameters()
and.named_parameters()
- How to collect the trainable parameters of a model
- How to use the
torchinfo
package (formerlytorch-summary
) to print a model summary
Checklist
- [X] The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
- [X] Only one issue is addressed in this pull request
- [X] Labels from the issue that this PR is fixing are added to this pull request
- [X] No unnecessary issues are included into this pull request.
cc @sekyondaMeta @svekars @kit1980 @brycebortree