fix for giannisnik#6

Accuracy = 0 as a result of recent PyTorch changes. See here for info: pytorch/pytorch#5411
robd2 · Sep 14, 2018 · 05acfc8 · 05acfc8
1 parent 18deed2
commit 05acfc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kcnn/main.py b/kcnn/main.py
@@ -106,7 +106,7 @@ def main():
 
         criterion = nn.CrossEntropyLoss()
 
-        evaluation = lambda output, target: torch.sum(output.eq(target)) / target.size()[0]
+        evaluation = lambda output, target: float (torch.sum(output.eq(target))) / float (target.size()[0])
 
         lr = args.lr
         lr_step = (args.lr-args.lr*args.lr_decay)/(args.epochs*args.schedule[1] - args.epochs*args.schedule[0])