ColossalAI/applications/ChatGPT/chatgpt/nn
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
..
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
actor.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
bloom_actor.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
bloom_critic.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
bloom_rm.py [chatgpt]support opt & gpt for rm training (#2876) 2023-02-22 16:58:11 +08:00
critic.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
generation.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
generation_utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
gpt_actor.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
gpt_critic.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
gpt_rm.py [chatgpt]support opt & gpt for rm training (#2876) 2023-02-22 16:58:11 +08:00
lora.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
loss.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
opt_actor.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
opt_critic.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
opt_rm.py [chatgpt]support opt & gpt for rm training (#2876) 2023-02-22 16:58:11 +08:00
reward_model.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