UltraLight-VM-UNet icon indicating copy to clipboard operation
UltraLight-VM-UNet copied to clipboard

Flops and param

Open fceex49 opened this issue 10 months ago • 1 comments

I have some doubts about the flops and params which are computed using thop. Because i think the Mamba layer is not supported there? This would lead zo wrong total flops and params?

fceex49 avatar Apr 20 '24 18:04 fceex49

Hi, there are still other ways to do the calculation of the parameters besides the issue 7 calculation:

def count_parameters(model):
    return sum(p.numel() for p in model.parameters() if p.requires_grad)

Thop is currently recognized as the most accurate way to calculate parameters and GFLOPs, including the same method of calculating parameters and GFLOPs provided in VM-UNet.

wurenkai avatar Apr 21 '24 01:04 wurenkai