How to re-train the model of BossNAS-M1 and M2

changlin31 / BossNAS

(ICCV 2021) BossNAS: Exploring Hybrid CNN-transformers with Block-wisely Self-supervised Neural Architecture Search

137 stars 20 forks source link

Closed Jxu-Thu closed 3 years ago

Jxu-Thu commented 3 years ago

Thanks for your great codes! Do you provide the codes to re-train the model on the MobileNet? such as BossNAS-M1 and M2

changlin31 commented 3 years ago

Thank you.

We use timm (https://github.com/rwightman/pytorch-image-models) to train our BossNAS-M models. Code for BossNAS-M models will be released in the future.