sail-sg / poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)
https://arxiv.org/abs/2111.11418
Apache License 2.0
1.3k stars 116 forks source link

How to check the number of parameters of both object detection and instance segmentation #48

Closed DoranLyong closed 1 year ago

DoranLyong commented 1 year ago

Thanks for your nice work. I'm employing your baseline for my work.

I have a question about how to get the number of parameters for object detection and instance segmentation in your paper. I checked how to do in image classification case in the 'misc', but I'm not familiar with mmlab's library. Could you please how to do that? Screenshot from 2023-02-07 19-57-27

yuweihao commented 1 year ago

Hi @DoranLyong , a simple method is to count the number of parameters stored in .pth file. Another method is to utilize tools offered by mmdet or mmseg like this.

DoranLyong commented 1 year ago

I've always appreciated your fast response! I solved it owing to your tips.