algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-27 12:03:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

change the resnet eval mode

Browse Source
This commit is contained in:
yunjey
2017-03-11 15:22:52 +09:00
parent 3ac1e9c2ef
commit 92aa5982f5
2 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tutorials/05 - Deep Residual Network/main-gpu.py
View File

@ -130,7 +130,6 @@ for epoch in range(80):
optimizer = torch.optim.Adam(resnet.parameters(), lr=lr) optimizer = torch.optim.Adam(resnet.parameters(), lr=lr)
# Test # Test
resnet.eval()
correct = 0 correct = 0
total = 0 total = 0
for images, labels in test_loader: for images, labels in test_loader:

1
tutorials/05 - Deep Residual Network/main.py
View File

@ -130,7 +130,6 @@ for epoch in range(80):
optimizer = torch.optim.Adam(resnet.parameters(), lr=lr) optimizer = torch.optim.Adam(resnet.parameters(), lr=lr)
# Test # Test
resnet.eval()
correct = 0 correct = 0
total = 0 total = 0
for images, labels in test_loader: for images, labels in test_loader: