search

huggingface / transfer-learning-conv-ai

🦄 State-of-the-Art Conversational AI with Transfer Learning
MIT License
1.74k stars 430 forks source link

Fine-tuning the model #96

Open ShivanshuPurohit opened 3 years ago

ShivanshuPurohit commented 3 years ago

Can I fine-tune this model for yet another dataset? By this I mean fine-tuning the model on yet another dataset, after having fine-tuned on the persona dataset.

nbertagnolli commented 3 years ago

I have some blog posts walking through this for two different datasets. Let me know if you still have trouble.

https://towardsdatascience.com/speak-to-the-dead-with-deep-learning-a336ef88425d

https://towardsdatascience.com/counsel-chat-bootstrapping-high-quality-therapy-data-971b419f33da