Giters

SHI-Labs / Neighborhood-Attention-Transformer

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to calculate the number of params?

HanzhouLiu opened this issue · comments

commented

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!

commented

Sorry. The pytorch built-in parameters does work.