mirror of https://github.com/hpcaitech/ColossalAI
75af66cd81
* hotfix/Fix get model policy strategy in ShardFormer * fix bug in auto policy |
||
---|---|---|
.. | ||
__init__.py | ||
shard_config.py | ||
sharder.py | ||
shardformer.py | ||
utils.py |