ColossalAI/applications/ChatGPT/chatgpt/trainer/strategies
ver217 78fd31f9c1
[chatgpt] add precision option for colossalai (#3233)
2023-03-24 12:15:06 +08:00
..
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
base.py change nn to models (#3032) 2023-03-07 16:34:22 +08:00
colossalai.py [chatgpt] add precision option for colossalai (#3233) 2023-03-24 12:15:06 +08:00
ddp.py [chatgpt] fix lora save bug (#3099) 2023-03-10 17:58:10 +08:00
naive.py [chatgpt] startegy add prepare method (#2766) 2023-02-17 11:27:27 +08:00
sampler.py [chatgpt] making experience support dp (#2971) 2023-03-03 15:51:19 +08:00