ColossalAI/applications/ChatGPT/chatgpt/trainer
ver217 4ee311c026
[chatgpt] startegy add prepare method (#2766)
* [chatgpt] startegy add prepare method

* [chatgpt] refactor examples

* [chatgpt] refactor strategy.prepare

* [chatgpt] support save/load checkpoint

* [chatgpt] fix unwrap actor

* [chatgpt] fix unwrap actor
2023-02-17 11:27:27 +08:00
..
callbacks [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
strategies [chatgpt] startegy add prepare method (#2766) 2023-02-17 11:27:27 +08:00
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
base.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
ppo.py [chatgpt] startegy add prepare method (#2766) 2023-02-17 11:27:27 +08:00
rm.py [chatgpt] support colossalai strategy to train rm (#2742) 2023-02-16 11:24:07 +08:00
utils.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00