Merge pull request #7 from robd2/master

fix for #6 Accuracy = 0 as a result of recent PyTorch changes. See here for info: pytorch/pytorch#5411
giannisnik · Feb 15, 2019 · 740c682 · 740c682
2 parents a63ef4b + 05acfc8
commit 740c682
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])