├── README.md ├── __init__.py ├── config └── config.json ├── data └── .gitignore ├── data_parallel.py ├── dataset.py ├── figure ├── clip-demo.jpg ├── gongzhonghao.jpeg └── model.png ├── generate_dialogue_subset.py ├── interact.py ├── interact_mmi.py ├── model └── .gitignore ├── preprocess.py ├── pytorchtools.py ├── requirements.txt ├── train.py └── vocab ├── vocab.txt └── vocab2.txt /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/config/config.json -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/data/.gitignore -------------------------------------------------------------------------------- /data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/data_parallel.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/dataset.py -------------------------------------------------------------------------------- /figure/clip-demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/figure/clip-demo.jpg -------------------------------------------------------------------------------- /figure/gongzhonghao.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/figure/gongzhonghao.jpeg -------------------------------------------------------------------------------- /figure/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/figure/model.png -------------------------------------------------------------------------------- /generate_dialogue_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/generate_dialogue_subset.py -------------------------------------------------------------------------------- /interact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/interact.py -------------------------------------------------------------------------------- /interact_mmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/interact_mmi.py -------------------------------------------------------------------------------- /model/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/model/.gitignore -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/preprocess.py -------------------------------------------------------------------------------- /pytorchtools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/pytorchtools.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers==4.4.2 2 | pytorch==1.7.0 3 | sklearn 4 | tqdm 5 | numpy 6 | scipy==1.2.1 -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/train.py -------------------------------------------------------------------------------- /vocab/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/vocab/vocab.txt -------------------------------------------------------------------------------- /vocab/vocab2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangjianxin1/GPT2-chitchat/HEAD/vocab/vocab2.txt --------------------------------------------------------------------------------