ColossalAI/colossalai/utils/model/colo_init_context.py at e070ca45c62e09fc709cd16bf77bd0bfef30eedc - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-16 17:16:14 +00:00

Files

Jiarui Fang c92f84fcdb [tensor] distributed checkpointing for parameters (#1240 )

2022-07-12 15:51:06 +08:00

3.6 KiB

Raw Blame History

View Raw