github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-18 16:00:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

[doc] update rst (#470)

Browse Source 
* update rst

* remove empty rst
This commit is contained in:
ver217
2022-03-21 10:52:45 +08:00
committed by GitHub
parent aff9d354f7
commit 7e30068a22
76 changed files with 454 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
docs/colossalai/colossalai.zero.rst
View File

@@ -4,10 +4,11 @@ colossalai.zero
.. automodule:: colossalai.zero
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.loss_scaler
colossalai.zero.zero_redundancy_optimizer_level_2
colossalai.zero.zero_redundancy_optimizer_level_3
colossalai.zero.init_ctx
colossalai.zero.shard_utils
colossalai.zero.sharded_model
colossalai.zero.sharded_optim
colossalai.zero.sharded_param