[model checkpoint] updated saving/loading for 2d layers (#595)

7636d518 · アマデウス · GitHub · cd13b638 · 7636d518
Unverified Commit 7636d518 authored Apr 01, 2022 by アマデウス Committed by GitHub Apr 01, 2022
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 575 additions and 8 deletions

colossalai/nn/layer/parallel_2d/layers.py colossalai/nn/layer/parallel_2d/layers.py +575 -8

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