ColossalAI/colossalai/interface
Hongxin Liu f51ce1bc8e [pipeline] refactor 1f1b schedule (#4115)
* [api] update optimizer wrapper to fit pipeline

* [pipeline] add base schedule

* [pipeline] add 1f1b schedule

* [test] add pipeline schedule utils test

* [pipeline] fix import
2023-08-15 23:25:14 +08:00
..
__init__.py [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
model.py [booster] implemented the torch ddd + resnet example (#3232) 2023-03-27 10:24:14 +08:00
optimizer.py [pipeline] refactor 1f1b schedule (#4115) 2023-08-15 23:25:14 +08:00