ColossalAI/applications/Chat/coati/trainer
Yuanchen 1ec0d386a9
reconstruct chat trainer and fix training script (#3588)
Co-authored-by: Yuanchen Xu <yuanchen.xu00@gmail.com>
2023-04-18 16:44:03 +08:00
..
callbacks [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00
strategies fix save_model inin naive and ddp strategy (#3436) 2023-04-04 15:30:01 +08:00
__init__.py [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00
base.py reconstruct chat trainer and fix training script (#3588) 2023-04-18 16:44:03 +08:00
ppo.py reconstruct chat trainer and fix training script (#3588) 2023-04-18 16:44:03 +08:00
rm.py reconstruct chat trainer and fix training script (#3588) 2023-04-18 16:44:03 +08:00
sft.py reconstruct chat trainer and fix training script (#3588) 2023-04-18 16:44:03 +08:00
utils.py [chatgpt] Detached PPO Training (#3195) 2023-04-17 14:46:50 +08:00