Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Op Flatten not supported #135

Open
MarkWijkhuizen opened this issue May 1, 2024 · 1 comment
Open

Op Flatten not supported #135

MarkWijkhuizen opened this issue May 1, 2024 · 1 comment
Labels
question Further information is requested

Comments

@MarkWijkhuizen
Copy link

When computing the macs/params using the get_model_complexity_info function I get the following message using Torch 2.2.1.

Op Flatten is not supported at now, set FLOPs of it to zero

Naturally, the Flatten operator is non-parameterized and should have 0 FLOPs.

Could you add the Flatten operator?

@sovrasov sovrasov added the question Further information is requested label May 6, 2024
@sovrasov
Copy link
Owner

sovrasov commented May 6, 2024

Flatten is a manipulation with memory (sometimes even copy is not required), and therefore it does not contribute to the overall number of macs/params.
The warnings are printed for further analysis, and for some operations they can be ignored.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants