tests/test_pipeline/test_middleware_1f1b.py · 09d69e1c25ac897f72e68782980948c26dfee324 · OpenDAS / ColossalAI

Ziyue Jiang authored Dec 12, 2022



* add bwd and step for PP middleware

* pre-commit
Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

09d69e1c

test_middleware_1f1b.py 4.58 KB