Skip to content

Op Flatten not supported #135

Open
Open
@MarkWijkhuizen

Description

@MarkWijkhuizen

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions