ColossalAI/applications/Chat/coati/trainer
Hongxin Liu 2a951955ad
[chat] refactor trainer (#3648)
* [chat] ppo trainer remove useless args

* [chat] update examples

* [chat] update benchmark

* [chat] update examples

* [chat] fix sft training with wandb

* [chat] polish docstr
2023-04-26 18:11:49 +08:00
..
callbacks [chat] polish performance evaluator (#3647) 2023-04-26 17:34:59 +08:00
strategies [gemini] accelerate inference (#3641) 2023-04-26 16:32:40 +08:00
__init__.py [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00
base.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
ppo.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
rm.py reconstruct chat trainer and fix training script (#3588) 2023-04-18 16:44:03 +08:00
sft.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
utils.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00