DialoGPT icon indicating copy to clipboard operation
DialoGPT copied to clipboard

How can I fine-tune the model on some subreddit?

Open btseytlin opened this issue 4 years ago • 1 comments

Thank you for this model, amazing results.

I would like to take the pretrained model and fine-tune it on r/advice. Such that it generates responses more similar to conversations in that subreddit.

How should I go about it?

btseytlin avatar Nov 13 '19 14:11 btseytlin

Here is my suggestion -- You might want to extract the subreddit data on your own, and follow our pipeline to create a dataset. Then, loading our model as initialization to fine-tune on the small dataset that you created by you own. However, you can be more creative than this :)

dreasysnail avatar Nov 22 '19 16:11 dreasysnail