ColossalAI/applications/Chat/coati/trainer
Baizhou Zhang 4da324cd60
[hotfix]fix argument naming in docs and examples (#4083)
2023-06-26 23:50:04 +08:00
..
callbacks [chat] polish performance evaluator (#3647) 2023-04-26 17:34:59 +08:00
strategies [hotfix]fix argument naming in docs and examples (#4083) 2023-06-26 23:50:04 +08:00
__init__.py
base.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
ppo.py [chat] refactor strategy class with booster api (#3987) 2023-06-25 17:36:21 +08:00
rm.py [chat] refactor strategy class with booster api (#3987) 2023-06-25 17:36:21 +08:00
sft.py [chat] refactor strategy class with booster api (#3987) 2023-06-25 17:36:21 +08:00
utils.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00