ColossalAI/applications/ChatGPT/chatgpt/trainer
BlueRum 3eebc4dff7
[chatgpt] fix rm eval (#2829)
* [chatgpt]fix train_rm bug with lora

* [chatgpt]support colossalai strategy to train rm

* fix pre-commit

* fix pre-commit 2

* [chatgpt]fix rm eval typo

* fix rm eval

* fix pre commit
2023-02-21 11:35:45 +08:00
..
callbacks [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
strategies [chatgpt] startegy add prepare method (#2766) 2023-02-17 11:27:27 +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] startegy add prepare method (#2766) 2023-02-17 11:27:27 +08:00
rm.py [chatgpt] fix rm eval (#2829) 2023-02-21 11:35:45 +08:00
utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00