How to fine tune using custom dataset
#2
by
KantimahantyRohit
- opened
I want to know how to fine tune this pre-trained model using my own dataset. I want to know that what should be the dataset format (columns). And since I am new to all these, a little help with the fine-tuning code will be really appreciated, regarding how to use Trainer to train this model.
Thanks
hi kantimahanty! I am on that too, did you find and answer?
You need two columns: ingredients and recipes. You can use the summarization.py script in the hugging face transformers repo. Instead of text and summary, you do ingredients and recipe
Could you share the link of that script?
guys can someone help
I wanna know how to train this model on my own data set