ColossalAI/colossalai.zero.sharded_param.rst at dev/gpt2_metainfo_patch - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 14:33:20 +00:00

ver217 d068af81a3

[doc] update rst and docstring (#1351 )

* update rst

* add zero docstr

* fix docstr

* remove fx.tracer.meta_patch

* fix docstr

* fix docstr

* update fx rst

* fix fx docstr

* remove useless rst

2022-07-21 15:54:53 +08:00

13 lines

249 B

ReStructuredText

Raw Permalink Blame History

 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