You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
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?The text was updated successfully, but these errors were encountered: