search

Project-MONAI / research-contributions

Implementations of recent research prototypes/demonstrations using MONAI.
https://monai.io/
Apache License 2.0
991 stars 326 forks source link

number of parameters (model Swin UNETR)? #58

Open Tombery1 opened 2 years ago

Tombery1 commented 2 years ago

Hi, I want to know, what is the number of parameters of model Swin UNETR?

ericspod commented 2 years ago

You can get the sum of trainable weights from a model with sum(p.numel() for p in model.parameters() if p.requires_grad) or using the torchinfo package to get a summary with that information.

Tombery1 commented 2 years ago

Thank you, I want to know how many parameters in the original paper, to compare the results with another model.

ericspod commented 2 years ago

@ahatamiz would you have the parameter count for your paper implementation? Thanks!

ahatamiz commented 2 years ago

Hi @ericspod and @Tombery1

The Swin UNETR model has 62M parameters. You can find this information in Table S.3. in this link. The official release also reports the number of parameters here.

Best,

TanThinNguyen commented 2 years ago

Hi @ahatamiz, besides the parameters, I see the FLOPs of Swin UNETR reported in the paper is 394.84G, would you release the code to calculate this? Thank you.