ColossalAI/applications/ChatGPT/chatgpt
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
..
dataset [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
experience_maker [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
nn [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
replay_buffer [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
trainer [chatgpt] support colossalai strategy to train rm (#2742) 2023-02-16 11:24:07 +08:00
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00