rm .data and .cpu

Scotter-Qian · Sep 18, 2018 · f6a4c4f · f6a4c4f
1 parent 733137f
commit f6a4c4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluation.py b/evaluation.py
@@ -43,7 +43,7 @@ def eval_loss(net, criterion, loader, use_cuda=False):
                 loss = criterion(outputs, targets)
                 total_loss += loss.item()*batch_size
                 _, predicted = torch.max(outputs.data, 1)
-                correct += predicted.eq(targets.data).cpu().sum().item()
+                correct += predicted.eq(targets).sum().item()
 
         elif isinstance(criterion, nn.MSELoss):
             for batch_idx, (inputs, targets) in enumerate(loader):