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

Skip to content

Commit 572dc13

Browse files
subramenbrianjo
andauthored
Remove final ReLU layer (#1654)
Co-authored-by: Brian Johnson <[email protected]>
1 parent 11df157 commit 572dc13

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

beginner_source/basics/quickstart_tutorial.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,7 @@ def __init__(self):
101101
nn.ReLU(),
102102
nn.Linear(512, 512),
103103
nn.ReLU(),
104-
nn.Linear(512, 10),
105-
nn.ReLU()
104+
nn.Linear(512, 10)
106105
)
107106

108107
def forward(self, x):

0 commit comments

Comments
 (0)