ColossalAI/docs/colossalai/colossalai.zero.shard_utils...

6 lines
186 B
ReStructuredText

colossalai.zero.shard\_utils.tensor\_shard\_strategy
====================================================
.. automodule:: colossalai.zero.shard_utils.tensor_shard_strategy
:members: