colossalai/booster/plugin/hybrid_parallel_plugin.py · 2a2ec49aa70c95f73b6017624e32cdad6b36b0e1 · OpenDAS / ColossalAI

[plugin]fix 3d checkpoint load when booster boost without optimizer. (#5135) · 2a2ec49a

flybird11111 authored Nov 30, 2023

* fix 3d checkpoint load when booster boost without optimizer

fix 3d checkpoint load when booster boost without optimizer

* test ci

* revert ci

* fix

fix

2a2ec49a

hybrid_parallel_plugin.py 55.5 KB

Replace hybrid_parallel_plugin.py