[model checkpoint] updated saving/loading for 3d layers (#597)

77ad24bf · アマデウス · GitHub · 93089ed7 · 77ad24bf
Unverified Commit 77ad24bf authored Apr 01, 2022 by アマデウス Committed by GitHub Apr 01, 2022
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 565 additions and 3 deletions

colossalai/nn/layer/parallel_3d/layers.py colossalai/nn/layer/parallel_3d/layers.py +565 -3

No files found.
--- a/colossalai/nn/layer/parallel_3d/layers.py
+++ b/colossalai/nn/layer/parallel_3d/layers.py