Commit e612f2e

authored

Merge pull request #268 from creeky123/master

Fix to accuracy function

2 parents 246c1fe + 7c1e3ca commit e612f2eCopy full SHA for e612f2e

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -434,7 +434,7 @@ def accuracy(output, target, topk=(1,)):`
`434`	`434`
`435`	`435`	`res = []`
`436`	`436`	`for k in topk:`
`437`		`- correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)`
	`437`	`+ correct_k = correct[:k].reshape(-1).float().sum(0, keepdim=True)`
`438`	`438`	`res.append(correct_k.mul_(100.0 / batch_size))`
`439`	`439`	`return res`
`440`	`440`

Comments

(0)