-
Notifications
You must be signed in to change notification settings - Fork 4.2k
Fix kernel dimensions for LeNet model code example #2192
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com>
✅ Deploy Preview for pytorch-tutorials-preview ready!
To edit notification comments on pull requests, go to your Netlify site settings. |
I'm just realizing that there's already an open issue for this: #2104. Apologies if I didn't give enough time for the issue's author to open a PR for it |
@suraj813 can you please take a look? Thanks! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@kiersten-stokes thanks for fixing this!
* Change kernel to 5x5 in 1st Conv2d layer in model init Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> * Change kernel to 5x5 in 2nd Conv2d layer in model init * Fix dimensions of 1st Linear layer to match new expected size --------- Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> Co-authored-by: Suraj Subramanian <5676233+suraj813@users.noreply.github.com> Co-authored-by: Svetlana Karslioglu <svekars@fb.com>
* Change kernel to 5x5 in 1st Conv2d layer in model init Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> * Change kernel to 5x5 in 2nd Conv2d layer in model init * Fix dimensions of 1st Linear layer to match new expected size --------- Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> Co-authored-by: Suraj Subramanian <5676233+suraj813@users.noreply.github.com> Co-authored-by: Svetlana Karslioglu <svekars@fb.com>
) * Fix kernel dimensions for LeNet model code example (#2192) * Change kernel to 5x5 in 1st Conv2d layer in model init Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> * Change kernel to 5x5 in 2nd Conv2d layer in model init * Fix dimensions of 1st Linear layer to match new expected size --------- Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> Co-authored-by: Suraj Subramanian <5676233+suraj813@users.noreply.github.com> Co-authored-by: Svetlana Karslioglu <svekars@fb.com> * Update pyspelling for all beginner tutorials in Python * Update * Update * Fix template tutorial, update gitignore * Apply suggestions from code review --------- Signed-off-by: Kiersten Stokes <kierstenstokes@gmail.com> Co-authored-by: Kiersten Stokes <kierstenstokes@gmail.com> Co-authored-by: Suraj Subramanian <5676233+suraj813@users.noreply.github.com> Co-authored-by: Nikita Shulga <nshulga@meta.com>
Fixes #2104
In going through the YouTube Intro tutorial, I noticed that the dimensions in the LeNet diagram don't match up with code that immediately follows it. Namely, the kernel size should be 5x5 rather than the 3x3 that's given. (And I believe the same should be true for the second convolutional layer?) While not really necessary for the point the tutorial is making, it may reduce confusion for those with no experience with this model (myself included 😄). Especially when comparing with the same network in this blitz tutorial that has the correct kernel size.
cc @suraj813