Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit e5176b6

Browse files
kiersten-stokessubramenSvetlana Karslioglu
authored
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 <[email protected]> * 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 <[email protected]> Co-authored-by: Suraj Subramanian <[email protected]> Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent 3764368 commit e5176b6

0 file changed

File tree

    0 commit comments

    Comments
     (0)