Remove unused argument for test method. (Taeung#746)

sim1st · Apr 11, 2020 · 80fcb66 · 80fcb66
1 parent 496d3b9
commit 80fcb66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mnist/main.py b/mnist/main.py
@@ -49,7 +49,7 @@ def train(args, model, device, train_loader, optimizer, epoch):
                 100. * batch_idx / len(train_loader), loss.item()))
 
 
-def test(args, model, device, test_loader):
+def test(model, device, test_loader):
     model.eval()
     test_loss = 0
     correct = 0
@@ -118,7 +118,7 @@ def main():
     scheduler = StepLR(optimizer, step_size=1, gamma=args.gamma)
     for epoch in range(1, args.epochs + 1):
         train(args, model, device, train_loader, optimizer, epoch)
-        test(args, model, device, test_loader)
+        test(model, device, test_loader)
         scheduler.step()
 
     if args.save_model: