ColossalAI/colossalai/nn/parallel/data_parallel.py at c6a1a626364316366bf155cfa125408f62fe3f55

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-11-30 15:55:09 +00:00

Files

HELSON c6a1a62636 [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786 )

* [hotfix] fix zero's incompatibility with checkpoint in torch-1.12

* [zero] add cpu shard init

* [zero] add tiny example test

* [colo_tensor] fix bugs for torch-1.11

2022-11-02 16:11:34 +08:00

25 KiB

Raw Blame History

View Raw

25 KiB Raw Blame History

25 KiB

Raw Blame History