mirror of https://github.com/hpcaitech/ColossalAI
14 lines
298 B
ReStructuredText
14 lines
298 B
ReStructuredText
colossalai.zero.sharded\_param
|
|
==============================
|
|
|
|
.. automodule:: colossalai.zero.sharded_param
|
|
:members:
|
|
|
|
|
|
.. toctree::
|
|
:maxdepth: 2
|
|
|
|
colossalai.zero.sharded_param.sharded_param
|
|
colossalai.zero.sharded_param.sharded_tensor
|
|
colossalai.zero.sharded_param.tensorful_state
|