ColossalAI/colossalai/pipeline
Hongxin Liu cd61353bae
[pipeline] hotfix backward for multiple outputs (#6090)
* [pipeline] hotfix backward for multiple outputs

* [pipeline] hotfix backward for multiple outputs
2024-10-16 17:27:33 +08:00
..
schedule [pipeline] hotfix backward for multiple outputs (#6090) 2024-10-16 17:27:33 +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