ColossalAI/colossalai/interface
Hongxin Liu 807e01a4ba
[zero] hotfix master param sync (#4618)
* [zero] add method to update master params

* [zero] update zero plugin

* [plugin] update low level zero plugin
2023-09-05 15:04:02 +08:00
..
__init__.py [zero] hotfix master param sync (#4618) 2023-09-05 15:04:02 +08:00
model.py [zero] hotfix master param sync (#4618) 2023-09-05 15:04:02 +08:00
optimizer.py [pipeline] refactor 1f1b schedule (#4115) 2023-08-15 23:25:14 +08:00