How do i fine-tune pretrained .pkl model ?

vlarine / ruDialoGPT

Training a Russian DialoGPT model.

MIT License

24 stars 4 forks source link

How do i fine-tune pretrained .pkl model ? #2

Open GraphGrailAi opened 4 years ago

GraphGrailAi commented 4 years ago

Hello, Vladimir Could you point some tutorial how do i fine-tune production-ready .pkl model 300K books dialogues model with a new dataset ? (not sure how to handle pkl, pth and load them)

shabashaash commented 4 years ago

.pkl - уже преобразованная просто тренируй её, надо только в train.sh пути прописать и всё. А вот если ты использовал так сказать "ванильную" ruGPT-2 тогда конвертируй.