yangsenius / TransPose

PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.
https://github.com/yangsenius/TransPose/releases/download/paper/transpose.pdf
MIT License
361 stars 58 forks source link

About n_head #22

Closed wangdong0556 closed 3 years ago

wangdong0556 commented 3 years ago

“For ResNet-S based, d=256, then n_heads = 8 = 256 // 32. For HRNet-S based, d=96, then n_heads = 1 = 96 // 96.” The values of RESNET and HRNet are 32 and 96 respectively. What are the meanings of these values(32 and 96)? Are they 96 for hrnet-s-w32 and w48?