zihangJiang / TokenLabeling

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"
Apache License 2.0
425 stars 36 forks source link

how to apply token labeling to CNN ? #29

Open HoJ00n2 opened 1 year ago

HoJ00n2 commented 1 year ago

Hello ~ I'm interested in your token labeling technique, So I want to apply this technique in CNN based model because ViT is very heavy to train.

can I get the your code with CNN token labeling? if you're not give me some detail for implementing

thank you.