fix se_resnext
@@ -333,7 +333,7 @@ def accuracy(output, target, topk=(1,)):
res = []
for k in topk:
- correct_k = correct[:k].view(-1).float().sum(0)
+ correct_k = correct[:k].reshape(-1).float().sum(0)
res.append(correct_k.mul_(100.0 / batch_size))
return res
