github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/applications/Chat/coati/trainer/strategies
History
Wenhao Chen edd75a59ea
[chat] remove naive strategy and split colossalai strategy (#4094) 
* feat: remove on_learn_epoch fn as not used

* revert: add _on_learn_epoch fn

* to: remove the use of NaiveStrategy

* test: remove NaiveStrategy tests

* feat: remove NaiveStrategy

* style: modify comments and params

* feat: split ColossalAIStrategy into LowLevelZeroStrategy and GeminiStrategy

* fix: remove naive

* fix: align with modified colossal strategy

* fix: fix ddp _try_init_dist arg
 		2023-06-29 18:11:00 +08:00
..
__init__.py [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00
base.py [chat] refactor strategy class with booster api (#3987) 2023-06-25 17:36:21 +08:00
colossalai.py [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00
ddp.py [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00
sampler.py [chat] add distributed PPO trainer (#3740) 2023-06-07 10:41:16 +08:00