Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

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

Merged
merged 3 commits into from
Aug 21, 2021

Conversation

linbinskn
Copy link
Contributor

When running amc example amc_search.py, error RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead. will be raised. Adding contiguous() to tensor can fix this bug.

@QuanluZhang QuanluZhang merged commit cae1e6d into microsoft:master Aug 21, 2021
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants