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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-27 02:03:13 +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