文件名称:GPT2 for Chinese chitchat/用于中文闲聊的GPT2模型-python
文件大小:169KB
文件格式:ZIP
更新时间:2024-07-08 13:59:25
自然语言处理
本项目使用GPT2模型对中文闲聊语料进行训练,使用 HuggingFace的transformers实现GPT2模型的编写与训练。
【文件预览】:
GPT2-chitchat-master
----vocab()
--------vocab2.txt(107KB)
--------vocab.txt(61KB)
----train.py(17KB)
----model()
--------.gitignore(68B)
----generate_dialogue_subset.py(3KB)
----requirements.txt(66B)
----dataset.py(479B)
----__init__.py(0B)
----pytorchtools.py(2KB)
----config()
--------config.json(874B)
----README.md(12KB)
----figure()
--------model.png(56KB)
----data()
--------.gitignore(68B)
----preprocess.py(4KB)
----interact_mmi.py(12KB)
----interact.py(9KB)
----data_parallel.py(4KB)