search

Haozhe-Xing / SARNet

SARNet
25 stars 3 forks source link

Code you use for the calculation of model Param, FLOPs #5

Open WkangLiu opened 1 year ago

WkangLiu commented 1 year ago

I would like to ask you which code you use for the calculation of model Param, FLOPs?

Can you recommend a github library?

I went to test it myself and it is half as high as yours, I think there is an error in my code, thanks!

Haozhe-Xing commented 1 year ago

from thop import profile As I know, a log of people use this package to calculate the FLOPs, you can also use this method.

WkangLiu commented 1 year ago

Thank you for your reply.

Haozhe-Xing commented 1 year ago

In addition, you can also test other models with your mehtods. Then you can compared with them.

Haozhe-Xing commented 1 year ago

In addition, you can also test other models with your methods. Then you can compared with them.

WkangLiu commented 1 year ago

OK，thank you, bro.