UltraLight-VM-UNet
UltraLight-VM-UNet copied to clipboard
Flops and param
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?
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.