From 92aa5982f58a9a073eba94915c6535cb7328823f Mon Sep 17 00:00:00 2001 From: yunjey Date: Sat, 11 Mar 2017 15:22:52 +0900 Subject: [PATCH] change the resnet eval mode --- tutorials/05 - Deep Residual Network/main-gpu.py | 1 - tutorials/05 - Deep Residual Network/main.py | 1 - 2 files changed, 2 deletions(-) diff --git a/tutorials/05 - Deep Residual Network/main-gpu.py b/tutorials/05 - Deep Residual Network/main-gpu.py index 2de3165..14cc3f7 100644 --- a/tutorials/05 - Deep Residual Network/main-gpu.py +++ b/tutorials/05 - Deep Residual Network/main-gpu.py @@ -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: diff --git a/tutorials/05 - Deep Residual Network/main.py b/tutorials/05 - Deep Residual Network/main.py index d849f88..aad0da2 100644 --- a/tutorials/05 - Deep Residual Network/main.py +++ b/tutorials/05 - Deep Residual Network/main.py @@ -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: