ColossalAI/applications/Chat/coati/trainer
Zirui Zhu 9e512938f6 [NFC] polish applications/Chat/coati/trainer/strategies/base.py code style (#4278) 2023-07-26 14:12:57 +08:00
..
callbacks [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00
strategies [NFC] polish applications/Chat/coati/trainer/strategies/base.py code style (#4278) 2023-07-26 14:12:57 +08:00
__init__.py [chat] refactor trainer class (#4080) 2023-06-29 10:48:09 +08:00
base.py [NFC] polish applications/Chat/coati/trainer/base.py code style (#4260) 2023-07-26 14:12:57 +08:00
ppo.py [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00
rm.py [chat] refactor trainer class (#4080) 2023-06-29 10:48:09 +08:00
sft.py [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00
utils.py [NFC] fix format of application/Chat/coati/trainer/utils.py (#4273) 2023-07-26 14:12:57 +08:00