Skip to content

Commit 92aa598

Browse files
committed
change the resnet eval mode
1 parent 3ac1e9c commit 92aa598

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

tutorials/05 - Deep Residual Network/main-gpu.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,6 @@ def forward(self, x):
130130
optimizer = torch.optim.Adam(resnet.parameters(), lr=lr)
131131

132132
# Test
133-
resnet.eval()
134133
correct = 0
135134
total = 0
136135
for images, labels in test_loader:

tutorials/05 - Deep Residual Network/main.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,6 @@ def forward(self, x):
130130
optimizer = torch.optim.Adam(resnet.parameters(), lr=lr)
131131

132132
# Test
133-
resnet.eval()
134133
correct = 0
135134
total = 0
136135
for images, labels in test_loader:

0 commit comments

Comments
 (0)