A GPT-2 ChatBot implemented using Pytorch and Huggingface-transformers

How to use

  1. install the required packages
pip install -r requirements.txt
  1. Download and process the data
  1. train the model

  you can train the model from a initial state

python --mode="train"

  also you can train a model from a specific checkpoint (eg. best.ckpt)

python --mode="train" --ckpt_name="best"
  1. infer & interact
python --mode="infer" --ckpt_name="best"


View Github