github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-14 03:56:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
208ac8f2ba67d8f43ec6f9024c3a4d112f9b4586
ColossalAI/colossalai/interface
History
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