fix for giannisnik#6

Accuracy = 0 as a result of recent PyTorch changes. See here for info: pytorch/pytorch#5411
robd2 · Sep 14, 2018 · e77488f · e77488f
1 parent d641f98
commit e77488f
Show file tree

Hide file tree

Showing 7 changed files with 1 addition and 1 deletion.
diff --git a/kcnn/__pycache__/graph_kernels.cpython-36.pyc b/kcnn/__pycache__/graph_kernels.cpython-36.pyc
diff --git a/kcnn/__pycache__/model.cpython-36.pyc b/kcnn/__pycache__/model.cpython-36.pyc
diff --git a/kcnn/__pycache__/nystrom.cpython-36.pyc b/kcnn/__pycache__/nystrom.cpython-36.pyc
diff --git a/kcnn/__pycache__/utils.cpython-36.pyc b/kcnn/__pycache__/utils.cpython-36.pyc
diff --git a/kcnn/checkpoint/kcnn/checkpoint.pth b/kcnn/checkpoint/kcnn/checkpoint.pth
diff --git a/kcnn/checkpoint/kcnn/model_best.pth b/kcnn/checkpoint/kcnn/model_best.pth
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])