Open
Description
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?