NEW: beginner recipe on layer parameters - Githubissues

pytorch / tutorials

PyTorch tutorials.

https://pytorch.org/tutorials/

BSD 3-Clause "New" or "Revised" License

7.96k stars 3.98k forks source link

NEW: beginner recipe on layer parameters #2948

Open loganthomas opened 1 week ago

loganthomas commented 1 week ago

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

pytorch-bot[bot] commented 1 week ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/2948

:page_facing_up: Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

:white_check_mark: No Failures

As of commit 71d17999555e25c6856c424c1ea9b06ff3224e85 with merge base 0740801fa8df2464f4d931d36a4bd7ada9cd0519 (): :green_heart: Looks good so far! There are no failures yet. :green_heart:

This comment was automatically generated by Dr. CI and updates every 15 minutes.