use reshape instead of view (#971)

a109f856 · ptrblck · GitHub · 4a1fa2c4 · a109f856
Unverified Commit a109f856 authored Sep 28, 2020 by ptrblck Committed by GitHub Sep 28, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/imagenet/main_amp.py examples/imagenet/main_amp.py +1 -1

No files found.
--- a/examples/imagenet/main_amp.py
+++ b/examples/imagenet/main_amp.py
@@ -527,7 +527,7 @@ def accuracy(output, target, topk=(1,)):

    res = []
    for k in topk:
-        correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)
+        correct_k = correct[:k].reshape(-1).float().sum(0, keepdim=True)
        res.append(correct_k.mul_(100.0 / batch_size))
    return res