[Model Compression] fix data contiguous bug in amc example (#4049)

cae1e6d4 · lin bin · GitHub · c0385c57 · cae1e6d4
Unverified Commit cae1e6d4 authored Aug 21, 2021 by lin bin Committed by GitHub Aug 21, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/model_compress/pruning/amc/utils.py examples/model_compress/pruning/amc/utils.py +1 -1

No files found.
--- a/examples/model_compress/pruning/amc/utils.py
+++ b/examples/model_compress/pruning/amc/utils.py
@@ -49,7 +49,7 @@ def accuracy(output, target, topk=(1, 5)):

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