ColossalAI/applications/ChatGPT/chatgpt/trainer
Fazzie-Maqianli bbac6760e5
fix torch version (#3225)
2023-03-23 20:56:35 +08:00
..
callbacks [chatgpt] Add saving ckpt callback for PPO (#2880) 2023-03-07 10:13:25 +08:00
strategies fix torch version (#3225) 2023-03-23 20:56:35 +08:00
__init__.py [chatgpt] add supervised learning fine-tune code (#3183) 2023-03-22 09:59:42 +08:00
base.py [chatgpt] making experience support dp (#2971) 2023-03-03 15:51:19 +08:00
ppo.py [chatgpt] fix trainer generate kwargs (#3166) 2023-03-17 17:31:22 +08:00
rm.py [chatgpt]Reward Model Training Process update (#3133) 2023-03-20 09:59:06 +08:00
sft.py [chatgpt] unnify datasets (#3218) 2023-03-23 17:38:30 +08:00
utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00