[model checkpoint] updated saving/loading for 2.5d layers (#596)

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

Showing with 537 additions and 7 deletions

colossalai/nn/layer/parallel_2p5d/layers.py colossalai/nn/layer/parallel_2p5d/layers.py +537 -7

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