github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-28 18:05:56 +00:00
Code Issues Packages Projects Releases Wiki Activity
f51ce1bc8e
ColossalAI/tests/test_pipeline/test_schedule
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
..
test_pipeline_schedule_utils.py [pipeline] refactor 1f1b schedule (#4115) 2023-08-15 23:25:14 +08:00