THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
681 stars 55 forks source link

fine-tuning dowstream classification task #36

Closed DAVID-Hown closed 5 months ago

DAVID-Hown commented 5 months ago

Hello author, thank you very much for your excellent work! I would like to ask for a classification task without using distillation, using the pre-trained model provided, which I set --finetune ./pretrained_weights/repvit_m2_3_distill_300e.pth \ --set_bn_eval NotImplementedError: Finetuning with distillation not yet supported? Is this operation not supported?