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 ChatGLM-6B:
python chat_6b.py