Fine tune on own column data

by mali17361 - opened Apr 25

Apr 25

Hi @nielsr . How Can I fine tune the model for only column detection ? The model is failing for some complex columns and I want to fine tune it own my own dataset. Can you plz suggest me with code how do I achieve it ? Also I want all 6 classes from the model after fine tuning the model.

nielsr

May 8

Yes that is supported, we just updated our object detection guide: https://huggingface.co/docs/transformers/main/en/tasks/object_detection, and we added example scripts for fine-tuning object detection models: https://github.com/huggingface/transformers/tree/main/examples/pytorch/object-detection.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment