Train to GPT-2 with own data
#73
by
JeisonJimenez
- opened
I would like to train gpt-2 with my documents to be able to ask questions about this information, what format should I give to my dataset?
Hi @JeisonJA
Thanks for your interest!
You might be interested in using SFTTrainer from TRL library: https://huggingface.co./docs/trl/sft_trainer let me know if you have any question