change the resnet eval mode

yunjey · yunjey · commit 92aa5982f58a · 2017-03-11T15:22:52.000+09:00
diff --git a/tutorials/05 - Deep Residual Network/main-gpu.py b/tutorials/05 - Deep Residual Network/main-gpu.py
@@ -130,7 +130,6 @@ def forward(self, x):
         optimizer = torch.optim.Adam(resnet.parameters(), lr=lr) 
         
 # Test
-resnet.eval()
 correct = 0
 total = 0
 for images, labels in test_loader:
diff --git a/tutorials/05 - Deep Residual Network/main.py b/tutorials/05 - Deep Residual Network/main.py
@@ -130,7 +130,6 @@ def forward(self, x):
         optimizer = torch.optim.Adam(resnet.parameters(), lr=lr) 
 
 # Test
-resnet.eval()
 correct = 0
 total = 0
 for images, labels in test_loader:
