github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 09:12:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ac178ca5c17ca751ff9df38be81da2b4a005fc0d
ColossalAI/colossalai/shardformer/__init__.py
FoolPlayer d3bc530849 [shardformer] Refactor shardformer api (#4001) 
* fix an error in readme

* simplify code

* refactor shardformer

* add todo

* remove slicer

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

2 lines
44 B
Python
Raw Blame History

from .shard import ShardConfig, ShardFormer
Reference in New Issue View Git Blame Copy Permalink