mirror of
https://github.com/yunjey/pytorch-tutorial.git
synced 2025-07-27 12:03:31 +08:00
change the resnet eval mode
This commit is contained in:
@ -130,7 +130,6 @@ for epoch in range(80):
|
||||
optimizer = torch.optim.Adam(resnet.parameters(), lr=lr)
|
||||
|
||||
# Test
|
||||
resnet.eval()
|
||||
correct = 0
|
||||
total = 0
|
||||
for images, labels in test_loader:
|
||||
|
@ -130,7 +130,6 @@ for epoch in range(80):
|
||||
optimizer = torch.optim.Adam(resnet.parameters(), lr=lr)
|
||||
|
||||
# Test
|
||||
resnet.eval()
|
||||
correct = 0
|
||||
total = 0
|
||||
for images, labels in test_loader:
|
||||
|
Reference in New Issue
Block a user