ColossalAI/applications/Chat/coati/trainer/callbacks
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 [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00
base.py [Coati] first commit (#3283) 2023-03-28 20:25:36 +08:00
performance_evaluator.py [chat] polish performance evaluator (#3647) 2023-04-26 17:34:59 +08:00
save_checkpoint.py [chat] remove naive strategy and split colossalai strategy (#4094) 2023-06-29 18:11:00 +08:00