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.

Sign up for GitHub

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

Jump to bottom

Training script #5

Open

zzadiues3338 opened this issue Apr 8, 2024 · 1 comment

zzadiues3338 commented Apr 8, 2024

Hi, I want to fine tune this model on my own dataset.
Do we have a training script for training the VegAnn model ?

Thanks a ton

Owner

simonMadec commented Apr 8, 2024

Yes here is an example of how we train Vegann with pytorch lighting :
https://github.com/simonMadec/VegAnn/blob/main/vegann/trainer.py

For fintening, you should load the model weights before or train by yourself on the VegAnn and then finetune!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment