Knover
Knover是一个基于PaddlePaddle的知识增强对话生成工具包。Knover允许研究人员和开发者高效地训练/推理大规模对话生成模型。
最新动态:
- 2022年10月:我们开放了Q-TOD,一种新颖的查询驱动任务型对话系统。
- 2022年3月:我们开放了PLATO-KAG,一种用于端到端知识增强对话建模的无监督学习方法。
- 2022年2月:我们在DSTC10-Track2中开放了TOD-DA数据集、模型和代码。
- 2021年12月:我们开放了PLATO-XL的对话生成模型,参数规模达110亿。
- 2021年10月:我们开放了AG-DST,一种用于对话状态追踪的可修正生成方法。
- 2021年2月:我们开放了在DSTC9-Track1中的实现(团队19)。
- 2020年7月:我们开放了PLATO-2,一个具有潜在空间的大规模生成模型,用于开放域对话系统。
要求和安装
- Python版本 >= 3.7
- paddlepaddle-gpu版本 >= 2.4.0
- 您可以按照说明安装PaddlePaddle。
- PaddlePaddle的具体版本还取决于您的CUDA版本(推荐版本:10.1)和CuDNN版本(推荐版本:7.6)。更多信息请参见PaddlePaddle关于GPU支持的文档
- sentencepiece
- termcolor
- 如果您想运行分布式训练,还需要NCCL
- 本地安装Knover:
git clone https://github.com/PaddlePaddle/Knover.git
cd Knover
pip3 install -e .
- 或者您可以仅设置
PYTHONPATH
:
export PYTHONPATH=/abs/path/to/Knover:$PYTHONPATH
基本用法
- 请参阅使用文档。
免责声明
本项目旨在促进对话生成领域的进一步研究进展。百度不对第三方使用预训练系统进行的生成负责。
联系方式
如需帮助或遇到使用Knover的问题,请提交GitHub issue。