Revert "fix for giannisnik#6"

This reverts commit e77488f.
robd2 · Sep 14, 2018 · 18deed2 · 18deed2
1 parent e77488f
commit 18deed2
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: float (torch.sum(output.eq(target))) / float (target.size()[0])
+        evaluation = lambda output, target: torch.sum(output.eq(target)) / 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])