Commit be85a0f3 authored by Frank Lee's avatar Frank Lee Committed by ver217
Browse files

removed tutorial markdown and refreshed rst files for consistency

parent ca4ae52d
colossalai.nn.loss.loss\_moe
============================
.. automodule:: colossalai.nn.loss.loss_moe
:members:
colossalai.nn.loss colossalai.nn.loss
================== ==================
.. toctree:: .. automodule:: colossalai.nn.loss
:maxdepth: 2 :members:
colossalai.nn.loss.cross_entropy_2d
colossalai.nn.loss.cross_entropy_2p5d
colossalai.nn.loss.cross_entropy_3d
.. toctree::
:maxdepth: 2
.. automodule:: colossalai.nn.loss colossalai.nn.loss.loss_2d
:members: colossalai.nn.loss.loss_2p5d
colossalai.nn.loss.loss_3d
colossalai.nn.loss.loss_moe
colossalai.nn.lr\_scheduler colossalai.nn.lr\_scheduler
=========================== ===========================
.. automodule:: colossalai.nn.lr_scheduler
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
...@@ -11,7 +15,3 @@ colossalai.nn.lr\_scheduler ...@@ -11,7 +15,3 @@ colossalai.nn.lr\_scheduler
colossalai.nn.lr_scheduler.onecycle colossalai.nn.lr_scheduler.onecycle
colossalai.nn.lr_scheduler.poly colossalai.nn.lr_scheduler.poly
colossalai.nn.lr_scheduler.torch colossalai.nn.lr_scheduler.torch
.. automodule:: colossalai.nn.lr_scheduler
:members:
colossalai.nn.metric.accuracy\_2d
=================================
.. automodule:: colossalai.nn.metric.accuracy_2d
:members:
colossalai.nn.metric.accuracy\_2p5d
===================================
.. automodule:: colossalai.nn.metric.accuracy_2p5d
:members:
colossalai.nn.metric.accuracy\_3d
=================================
.. automodule:: colossalai.nn.metric.accuracy_3d
:members:
colossalai.nn.metric
====================
.. automodule:: colossalai.nn.metric
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.metric.accuracy_2d
colossalai.nn.metric.accuracy_2p5d
colossalai.nn.metric.accuracy_3d
colossalai.nn.model.model\_from\_config colossalai.nn.model.model\_from\_config
=============================== =======================================
.. automodule:: colossalai.nn.model.model_from_config .. automodule:: colossalai.nn.model.model_from_config
:members: :members:
colossalai.nn.model colossalai.nn.model
=================== ===================
.. automodule:: colossalai.nn.model
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
......
colossalai.nn.optimizer.colossalai\_optimizer
=============================================
.. automodule:: colossalai.nn.optimizer.colossalai_optimizer
:members:
colossalai.nn.optimizer colossalai.nn.optimizer
======================= =======================
.. automodule:: colossalai.nn.optimizer
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.nn.optimizer.colossalai_optimizer
colossalai.nn.optimizer.fused_adam colossalai.nn.optimizer.fused_adam
colossalai.nn.optimizer.fused_lamb colossalai.nn.optimizer.fused_lamb
colossalai.nn.optimizer.fused_sgd colossalai.nn.optimizer.fused_sgd
colossalai.nn.optimizer.lamb colossalai.nn.optimizer.lamb
colossalai.nn.optimizer.lars colossalai.nn.optimizer.lars
.. automodule:: colossalai.nn.optimizer
:members:
colossalai.nn colossalai.nn
============= =============
.. automodule:: colossalai.nn
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.nn.layer colossalai.nn.layer
colossalai.nn.loss colossalai.nn.loss
colossalai.nn.lr_scheduler colossalai.nn.lr_scheduler
colossalai.nn.metric
colossalai.nn.model colossalai.nn.model
colossalai.nn.optimizer colossalai.nn.optimizer
.. automodule:: colossalai.nn .. toctree::
:members: :maxdepth: 2
colossalai.nn.init
colossalai.registry colossalai.registry
=================== ===================
.. automodule:: colossalai.registry
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.registry.registry colossalai.registry.registry
.. automodule:: colossalai.registry
:members:
colossalai colossalai
========== ==========
.. toctree:: .. automodule:: colossalai
:maxdepth: 2 :members:
colossalai.constants
colossalai.core
colossalai.initialize
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
...@@ -17,6 +12,7 @@ colossalai ...@@ -17,6 +12,7 @@ colossalai
colossalai.communication colossalai.communication
colossalai.context colossalai.context
colossalai.engine colossalai.engine
colossalai.kernel
colossalai.logging colossalai.logging
colossalai.nn colossalai.nn
colossalai.registry colossalai.registry
...@@ -24,5 +20,8 @@ colossalai ...@@ -24,5 +20,8 @@ colossalai
colossalai.utils colossalai.utils
colossalai.zero colossalai.zero
.. automodule:: colossalai
:members: .. toctree::
:maxdepth: 2
colossalai.initialize
colossalai.trainer.metric
=========================
.. automodule:: colossalai.trainer.metric
:members:
colossalai.trainer colossalai.trainer
================== ==================
.. toctree:: .. automodule:: colossalai.trainer
:maxdepth: 2 :members:
colossalai.trainer.hooks
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.trainer.metric colossalai.trainer.hooks
.. automodule:: colossalai.trainer
:members:
colossalai.utils.data\_sampler.base\_sampler
============================================
.. automodule:: colossalai.utils.data_sampler.base_sampler
:members:
colossalai.utils.data\_sampler.data\_parallel\_sampler
======================================================
.. automodule:: colossalai.utils.data_sampler.data_parallel_sampler
:members:
colossalai.utils.data\_sampler colossalai.utils.data\_sampler
======================================= ==============================
.. automodule:: colossalai.utils.data_sampler .. automodule:: colossalai.utils.data_sampler
:members: :members:
.. toctree::
:maxdepth: 2
colossalai.utils.data_sampler.base_sampler
colossalai.utils.data_sampler.data_parallel_sampler
colossalai.utils.multi\_tensor\_apply.multi\_tensor\_apply
==========================================================
.. automodule:: colossalai.utils.multi_tensor_apply.multi_tensor_apply
:members:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment