mirror of https://github.com/hpcaitech/ColossalAI
edd75a59ea
* 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 |
||
|---|---|---|
| .. | ||
| callbacks | ||
| strategies | ||
| __init__.py | ||
| base.py | ||
| ppo.py | ||
| rm.py | ||
| sft.py | ||
| utils.py | ||