github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/applications/ChatGPT/chatgpt/trainer
History
BlueRum 613efebc5c
[chatgpt] support colossalai strategy to train rm (#2742) 
* [chatgpt]fix train_rm bug with lora

* [chatgpt]support colossalai strategy to train rm

* fix pre-commit

* fix pre-commit 2
 		2023-02-16 11:24:07 +08:00
..
callbacks [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
strategies [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
base.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
ppo.py [chatgpt] optimize generation kwargs (#2717) 2023-02-15 13:59:58 +08:00
rm.py [chatgpt] support colossalai strategy to train rm (#2742) 2023-02-16 11:24:07 +08:00
utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00