README.md

How to Use

Users can train the dialog task via:

python train_ppo.py --config nlp_ppo.yaml

Users can train the dialog task with deepspeed via:

deepspeed train_ppo.py --config nlp_ppo_ds.yaml

After the training, users can chat with the agent via:

python chat.py

Chat with other agents

• Chat with ChatGLM-6B: python chat_6b.py