-
Notifications
You must be signed in to change notification settings - Fork 216
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Recommendations for configuring heads/training on custom datasets? #1
Comments
do you think you could share the jupyter notebook for training? |
@lessw2020 Inspired by this issue, we have added the functionality of fine-tuning pre-trained classification backbones on other classification datasets in v0.2 of this library. See fine-tuning example here. |
Closing issue because of no activity |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks for developing MobileVit!
I'm wondering if there are any specific tips/examples for fine-tuning the pre-trained classification and detection models using mobilevit on custom datasets?
I see the n_classes reference in both classifier (1000) and detection (80), but can you provide any quick example of modifying for custom datasets and if you have any recommended lr for finetuning?
Thanks very much!
The text was updated successfully, but these errors were encountered: