学习笔记(50)- DialoGPT
A State-of-the-Art Large-scale Pretrained Response Generation Model
https://github.com/microsoft/DialoGPT
仓库基于 huggingface 的 pytorch-transformer 和 OpenAI 的 GPT-2
2个数据集:DSTC-7 、6K multi-ref dataset from Reddit(数据集未公布)
模型文件117M 、345M、 762M均可下载。
提供了评测脚本:rouge、meteor、cider、bleu.