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
==================
.. toctree::
:maxdepth: 2
.. automodule:: colossalai.nn.loss
: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
:members:
colossalai.nn.loss.loss_2d
colossalai.nn.loss.loss_2p5d
colossalai.nn.loss.loss_3d
colossalai.nn.loss.loss_moe
colossalai.nn.lr\_scheduler
===========================
.. automodule:: colossalai.nn.lr_scheduler
:members:
.. toctree::
:maxdepth: 2
......@@ -11,7 +15,3 @@ colossalai.nn.lr\_scheduler
colossalai.nn.lr_scheduler.onecycle
colossalai.nn.lr_scheduler.poly
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
===============================
=======================================
.. automodule:: colossalai.nn.model.model_from_config
:members:
colossalai.nn.model
===================
.. automodule:: colossalai.nn.model
:members:
.. toctree::
:maxdepth: 2
......
colossalai.nn.optimizer.colossalai\_optimizer
=============================================
.. automodule:: colossalai.nn.optimizer.colossalai_optimizer
:members:
colossalai.nn.optimizer
=======================
.. automodule:: colossalai.nn.optimizer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.optimizer.colossalai_optimizer
colossalai.nn.optimizer.fused_adam
colossalai.nn.optimizer.fused_lamb
colossalai.nn.optimizer.fused_sgd
colossalai.nn.optimizer.lamb
colossalai.nn.optimizer.lars
.. automodule:: colossalai.nn.optimizer
:members:
colossalai.nn
=============
.. automodule:: colossalai.nn
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer
colossalai.nn.loss
colossalai.nn.lr_scheduler
colossalai.nn.metric
colossalai.nn.model
colossalai.nn.optimizer
.. automodule:: colossalai.nn
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.init
colossalai.registry
===================
.. automodule:: colossalai.registry
:members:
.. toctree::
:maxdepth: 2
colossalai.registry.registry
.. automodule:: colossalai.registry
:members:
colossalai
==========
.. toctree::
:maxdepth: 2
colossalai.constants
colossalai.core
colossalai.initialize
.. automodule:: colossalai
:members:
.. toctree::
:maxdepth: 2
......@@ -17,6 +12,7 @@ colossalai
colossalai.communication
colossalai.context
colossalai.engine
colossalai.kernel
colossalai.logging
colossalai.nn
colossalai.registry
......@@ -24,5 +20,8 @@ colossalai
colossalai.utils
colossalai.zero
.. automodule:: colossalai
:members:
.. toctree::
:maxdepth: 2
colossalai.initialize
colossalai.trainer.metric
=========================
.. automodule:: colossalai.trainer.metric
:members:
colossalai.trainer
==================
.. toctree::
:maxdepth: 2
colossalai.trainer.hooks
.. automodule:: colossalai.trainer
:members:
.. toctree::
:maxdepth: 2
colossalai.trainer.metric
.. automodule:: colossalai.trainer
:members:
colossalai.trainer.hooks
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
=======================================
==============================
.. automodule:: colossalai.utils.data_sampler
: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