search

hustvl / TopFormer

TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022
Other
373 stars 42 forks source link

Classification cfg? #5

Closed xperzy closed 2 years ago

xperzy commented 2 years ago

Hello, could you please provide the cfgs (e.g., classification head) for classification model?

Thanks.

mulinmeng commented 2 years ago

Hello, could you please provide the cfgs (e.g., classification head) for classification model?

Thanks.

Thanks for your ask! You can view specific settings from the classification checkpoint.

The classification head is a Linear-BN layer. The detailed structure is shown in Figure 5 of the arxiv paper.