Does it support fine-tuning?
#3
by
mserrasa
- opened
I want to fine-tune this model to a specific dataset. I know hustvl/yolos-tiny does support it but i would like to try this one since it achieves better results.
Yes it does, we just updated our fine-tuning guides, which includes support for YOLOS!
Guide: https://huggingface.co/docs/transformers/en/tasks/object_detection
Official scripts (using either the Trainer API or Accelerate): https://github.com/huggingface/transformers/tree/main/examples/pytorch/object-detection