ColossalAI/applications/ChatGPT/chatgpt/trainer
BlueRum c9dd036592
[chatgpt] fix lora save bug (#3099)
* fix colo-stratergy

* polish

* fix lora

* fix ddp

* polish

* polish
2023-03-10 17:58:10 +08:00
..
callbacks [chatgpt] Add saving ckpt callback for PPO (#2880) 2023-03-07 10:13:25 +08:00
strategies [chatgpt] fix lora save bug (#3099) 2023-03-10 17:58:10 +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 change nn to models (#3032) 2023-03-07 16:34:22 +08:00
rm.py change nn to models (#3032) 2023-03-07 16:34:22 +08:00
utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00