Add a bugfix for conv1d in the cdarts utils. In utils.py, change accu…

…racy's view to a reshape. (#3073)
microsoft · Nov 13, 2020 · fcbf05b · fcbf05b
1 parent 26b71c4
commit fcbf05b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nni/algorithms/nas/pytorch/cdarts/utils.py b/nni/algorithms/nas/pytorch/cdarts/utils.py
@@ -58,7 +58,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_(1.0 / batch_size))
     return res