How to calculate the number of params?

SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022

MIT License

1.04k stars 85 forks source link

How to calculate the number of params? #95

Closed HanzhouLiu closed 9 months ago

HanzhouLiu commented 9 months ago

Thanks for your great work!! However, I am still kind of confused how to calculate #params using the pip package. The pytorch built-in model.parameters does not work here because the parameters of natten are ignored. Any idea would be appreciated!

HanzhouLiu commented 9 months ago

Sorry. The pytorch built-in parameters does work.