Fix Parameter Server Example (Taeung#774)

sim1st · May 21, 2020 · 31643b2 · 31643b2
1 parent e9b2f8e
commit 31643b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/rpc/parameter_server/rpc_parameter_server.py b/distributed/rpc/parameter_server/rpc_parameter_server.py
@@ -202,7 +202,7 @@ def get_accuracy(test_loader, model):
         and torch.cuda.is_available() else "cpu")
     with torch.no_grad():
         for i, (data, target) in enumerate(test_loader):
-            out = model(data, -1)
+            out = model(data)
             pred = out.argmax(dim=1, keepdim=True)
             pred, target = pred.to(device), target.to(device)
             correct = pred.eq(target.view_as(pred)).sum().item()