BEVFormer icon indicating copy to clipboard operation
BEVFormer copied to clipboard

Calculate FLOPs

Open VincentWangty opened this issue 3 years ago • 3 comments

It's a really great study. Could you share your code to calculate the FLOPs. I try to use get_model_complexity_info() to calculate, however, the shape is not right.

VincentWangty avatar Jul 25 '22 06:07 VincentWangty

You should provide a dummy_metas for forward_dummy in https://github.com/zhiqi-li/BEVFormer/blob/master/projects/mmdet3d_plugin/bevformer/detectors/bevformer.py#L140. Then you can use get_flops of mmdet3d to obtain the FLOPs

zhiqi-li avatar Jul 25 '22 13:07 zhiqi-li

Thanks for your replay!

VincentWangty avatar Jul 26 '22 02:07 VincentWangty

Thanks for your replay!

Hello, thank for your work, but i want to know what the input_shape is, or which command do you run? can you show it for me? thx

admyxs avatar Jan 09 '24 03:01 admyxs