html refactor (#555)

docs/colossalai/colossalai.utils.tensor_detector.tensor_detector.rst

-colossalai.utils.tensor\_detector.tensor\_detector
-==================================================
-
-.. automodule:: colossalai.utils.tensor_detector.tensor_detector
-   :members:

docs/colossalai/colossalai.zero.init_ctx.init_context.rst

-colossalai.zero.init\_ctx.init\_context
-=======================================
-
-.. automodule:: colossalai.zero.init_ctx.init_context
-   :members:

docs/colossalai/colossalai.zero.init_ctx.rst

 colossalai.zero.init\_ctx
 =========================
 
-.. automodule:: colossalai.zero.init_ctx
-   :members:
-
+colossalai.zero.init\_ctx.init\_context
+-----------------------------------------
 
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.zero.init_ctx.init_context
+.. automodule:: colossalai.zero.init_ctx.init_context
+   :members:

docs/colossalai/colossalai.zero.shard_utils.base_shard_strategy.rst

-colossalai.zero.shard\_utils.base\_shard\_strategy
-==================================================
-
-.. automodule:: colossalai.zero.shard_utils.base_shard_strategy
-   :members:

docs/colossalai/colossalai.zero.shard_utils.bucket_tensor_shard_strategy.rst

-colossalai.zero.shard\_utils.bucket\_tensor\_shard\_strategy
-============================================================
-
-.. automodule:: colossalai.zero.shard_utils.bucket_tensor_shard_strategy
-   :members:

docs/colossalai/colossalai.zero.shard_utils.rst

@@ -3,11 +3,3 @@ colossalai.zero.shard\_utils
 
 .. automodule:: colossalai.zero.shard_utils
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.zero.shard_utils.base_shard_strategy
-   colossalai.zero.shard_utils.bucket_tensor_shard_strategy
-   colossalai.zero.shard_utils.tensor_shard_strategy

docs/colossalai/colossalai.zero.shard_utils.tensor_shard_strategy.rst

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

docs/colossalai/colossalai.zero.sharded_model.param_manager.rst

-colossalai.zero.sharded\_model.param\_manager
-=============================================
-
-.. automodule:: colossalai.zero.sharded_model.param_manager
-   :members:

docs/colossalai/colossalai.zero.sharded_model.reduce_scatter.rst

-colossalai.zero.sharded\_model.reduce\_scatter
-==============================================
-
-.. automodule:: colossalai.zero.sharded_model.reduce_scatter
-   :members:

docs/colossalai/colossalai.zero.sharded_model.rst

 colossalai.zero.sharded\_model
 ==============================
 
+
+colossalai.zero.sharded\_model.sharded\_model\_v2
+--------------------------------------------------
+
+.. automodule:: colossalai.zero.sharded_model.sharded_model_v2
+   :members:
+
+colossalai.zero.sharded\_model.reduce\_scatter
+-----------------------------------------------
+
+.. automodule:: colossalai.zero.sharded_model.reduce_scatter
+   :members:
+
+colossalai.zero.sharded\_model.utils
+--------------------------------------
+
+.. automodule:: colossalai.zero.sharded_model.utils
+   :members:
+
+colossalai.zero.sharded\_model.sharded\_model
+--------------------------------------------------
+
 .. automodule:: colossalai.zero.sharded_model
    :members:
 
+colossalai.zero.sharded\_model.sharded\_grad
+--------------------------------------------------
+
+.. automodule:: colossalai.zero.sharded_grad
+   :members:
 
-.. toctree::
-   :maxdepth: 2
+colossalai.zero.sharded\_model.param\_manager
+----------------------------------------------
 
-   colossalai.zero.sharded_model.param_manager
-   colossalai.zero.sharded_model.reduce_scatter
-   colossalai.zero.sharded_model.sharded_grad
-   colossalai.zero.sharded_model.sharded_model
-   colossalai.zero.sharded_model.sharded_model_v2
-   colossalai.zero.sharded_model.utils
+.. automodule:: colossalai.zero.sharded_model.param_manager
+   :members:
\ No newline at end of file

docs/colossalai/colossalai.zero.sharded_model.sharded_grad.rst

-colossalai.zero.sharded\_model.sharded\_grad
-============================================
-
-.. automodule:: colossalai.zero.sharded_model.sharded_grad
-   :members:

docs/colossalai/colossalai.zero.sharded_model.sharded_model.rst

-colossalai.zero.sharded\_model.sharded\_model
-=============================================
-
-.. automodule:: colossalai.zero.sharded_model.sharded_model
-   :members:

docs/colossalai/colossalai.zero.sharded_model.sharded_model_v2.rst

-colossalai.zero.sharded\_model.sharded\_model\_v2
-=================================================
-
-.. automodule:: colossalai.zero.sharded_model.sharded_model_v2
-   :members:

docs/colossalai/colossalai.zero.sharded_model.utils.rst

-colossalai.zero.sharded\_model.utils
-====================================
-
-.. automodule:: colossalai.zero.sharded_model.utils
-   :members:

docs/colossalai/colossalai.zero.sharded_optim.bookkeeping.base_store.rst

-colossalai.zero.sharded\_optim.bookkeeping.base\_store
-======================================================
-
-.. automodule:: colossalai.zero.sharded_optim.bookkeeping.base_store
-   :members:

docs/colossalai/colossalai.zero.sharded_optim.bookkeeping.bucket_store.rst

-colossalai.zero.sharded\_optim.bookkeeping.bucket\_store
-========================================================
-
-.. automodule:: colossalai.zero.sharded_optim.bookkeeping.bucket_store
-   :members:

docs/colossalai/colossalai.zero.sharded_optim.bookkeeping.gradient_store.rst

-colossalai.zero.sharded\_optim.bookkeeping.gradient\_store
-==========================================================
-
-.. automodule:: colossalai.zero.sharded_optim.bookkeeping.gradient_store
-   :members:

docs/colossalai/colossalai.zero.sharded_optim.bookkeeping.parameter_store.rst

-colossalai.zero.sharded\_optim.bookkeeping.parameter\_store
-===========================================================
-
-.. automodule:: colossalai.zero.sharded_optim.bookkeeping.parameter_store
-   :members:

docs/colossalai/colossalai.zero.sharded_optim.bookkeeping.rst

-colossalai.zero.sharded\_optim.bookkeeping
-==========================================
-
-.. automodule:: colossalai.zero.sharded_optim.bookkeeping
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.zero.sharded_optim.bookkeeping.base_store
-   colossalai.zero.sharded_optim.bookkeeping.bucket_store
-   colossalai.zero.sharded_optim.bookkeeping.gradient_store
-   colossalai.zero.sharded_optim.bookkeeping.parameter_store
-   colossalai.zero.sharded_optim.bookkeeping.tensor_bucket

docs/colossalai/colossalai.zero.sharded_optim.bookkeeping.tensor_bucket.rst

-colossalai.zero.sharded\_optim.bookkeeping.tensor\_bucket
-=========================================================
-
-.. automodule:: colossalai.zero.sharded_optim.bookkeeping.tensor_bucket
-   :members: