ColossalAI/colossalai/pipeline
Hongxin Liu 13946c4448
[fp8] hotfix backward hook (#6053)
* [fp8] hotfix backward hook

* [fp8] hotfix pipeline loss accumulation
2024-09-11 16:11:25 +08:00
..
schedule [fp8] hotfix backward hook (#6053) 2024-09-11 16:11:25 +08:00
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
p2p.py [FP8] rebase main (#5963) 2024-08-06 16:29:37 +08:00
stage_manager.py [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00