Morph MORPH
SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 38 Python

[shardformer] refactor pipeline grad ckpt config (#5646)

* [shardformer] refactor pipeline grad ckpt config

* [shardformer] refactor pipeline grad ckpt config

* [pipeline] fix stage manager
H
Hongxin Liu committed
1b387ca9fe2fe7f90459537b0cc19d5bb4edbdc5
Parent: 7ef9160
Committed by GitHub <noreply@github.com> on 4/25/2024, 7:19:30 AM