github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 01:19:15 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
92f6791095491e44c5712e14f00f2e19b52dc9f6
ColossalAI/test.py
Frank Lee 8eb09a4c69 [shardformer] support module saving and loading (#4062) 
* [shardformer] support module saving and loading

* polish code
2023-07-04 16:05:01 +08:00

2 lines
65 B
Python
Raw Blame History

from colossalai.tensor.d_tensor.api import to_distributed_tensor
Reference in New Issue View Git Blame Copy Permalink