ColossalAI/applications/ChatGPT/chatgpt/trainer
BlueRum f5ca0397dd
[chatgpt] fix lora gemini conflict in RM training (#2984)
* fix lora bug

* polish

* fix lora gemini
2023-03-03 15:58:16 +08:00
..
callbacks [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
strategies [chatgpt] making experience support dp (#2971) 2023-03-03 15:51:19 +08:00
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
base.py [chatgpt] making experience support dp (#2971) 2023-03-03 15:51:19 +08:00
ppo.py [chatgpt] startegy add prepare method (#2766) 2023-02-17 11:27:27 +08:00
rm.py [chatgpt] fix lora gemini conflict in RM training (#2984) 2023-03-03 15:58:16 +08:00
utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00