github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-24 21:12:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
8eb09a4c6946b40930cffc7f2d9bb150ee714b63
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