tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard

NEW: beginner recipe on layer parameters

Open loganthomas opened this issue 8 months ago • 1 comments

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 (formerly torch-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

loganthomas avatar Jun 19 '24 20:06 loganthomas