ColossalAI/applications/Chat/coati/trainer
tanitna 1a60dc07a8
[chat] typo accimulation_steps -> accumulation_steps (#3662)
2023-04-28 15:42:57 +08:00
..
callbacks [chat] polish performance evaluator (#3647) 2023-04-26 17:34:59 +08:00
strategies [chat] refactor model save/load logic (#3654) 2023-04-27 18:41:49 +08:00
__init__.py
base.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00
ppo.py [chat] refactor model save/load logic (#3654) 2023-04-27 18:41:49 +08:00
rm.py [chat] refactor model save/load logic (#3654) 2023-04-27 18:41:49 +08:00
sft.py [chat] typo accimulation_steps -> accumulation_steps (#3662) 2023-04-28 15:42:57 +08:00
utils.py [chat] refactor trainer (#3648) 2023-04-26 18:11:49 +08:00