Commit 404ecbdc authored by zbian's avatar zbian
Browse files

Migrated project

parent 2ebaefc5
colossalai.nn.layer.parallel\_vision\_transformer.layers
========================================================
.. automodule:: colossalai.nn.layer.parallel_vision_transformer.layers
:members:
colossalai.nn.layer.parallel\_vision\_transformer
=================================================
.. automodule:: colossalai.nn.layer.parallel_vision_transformer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_vision_transformer.layers
colossalai.nn.layer
===================
.. automodule:: colossalai.nn.layer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_1d
colossalai.nn.layer.parallel_2d
colossalai.nn.layer.parallel_2p5d
colossalai.nn.layer.parallel_3d
colossalai.nn.layer.parallel_sequence
colossalai.nn.layer.parallel_vision_transformer
colossalai.nn.layer.vanilla_resnet
colossalai.nn.layer.vanilla_vision_transformer
colossalai.nn.layer.wrapper
.. toctree::
:maxdepth: 2
colossalai.nn.layer.base_layer
colossalai.nn.layer.vanilla\_resnet.basic\_block
================================================
.. automodule:: colossalai.nn.layer.vanilla_resnet.basic_block
:members:
colossalai.nn.layer.vanilla\_resnet.bottleneck
==============================================
.. automodule:: colossalai.nn.layer.vanilla_resnet.bottleneck
:members:
colossalai.nn.layer.vanilla\_resnet.conv
========================================
.. automodule:: colossalai.nn.layer.vanilla_resnet.conv
:members:
colossalai.nn.layer.vanilla\_resnet.reslayer
============================================
.. automodule:: colossalai.nn.layer.vanilla_resnet.reslayer
:members:
colossalai.nn.layer.vanilla\_resnet
===================================
.. automodule:: colossalai.nn.layer.vanilla_resnet
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.vanilla_resnet.basic_block
colossalai.nn.layer.vanilla_resnet.bottleneck
colossalai.nn.layer.vanilla_resnet.conv
colossalai.nn.layer.vanilla_resnet.reslayer
colossalai.nn.layer.vanilla\_vision\_transformer.layers
=======================================================
.. automodule:: colossalai.nn.layer.vanilla_vision_transformer.layers
:members:
colossalai.nn.layer.vanilla\_vision\_transformer
================================================
.. automodule:: colossalai.nn.layer.vanilla_vision_transformer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.vanilla_vision_transformer.layers
colossalai.nn.layer.wrapper.lambda\_wrapper
===========================================
.. automodule:: colossalai.nn.layer.wrapper.lambda_wrapper
:members:
colossalai.nn.layer.wrapper
===========================
.. automodule:: colossalai.nn.layer.wrapper
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.wrapper.lambda_wrapper
colossalai.nn.loss.base\_loss
=============================
.. automodule:: colossalai.nn.loss.base_loss
:members:
colossalai.nn.loss.cross\_entropy\_1d
=====================================
.. automodule:: colossalai.nn.loss.cross_entropy_1d
:members:
colossalai.nn.loss.cross\_entropy\_2d
=====================================
.. automodule:: colossalai.nn.loss.cross_entropy_2d
:members:
colossalai.nn.loss.cross\_entropy\_2p5d
=======================================
.. automodule:: colossalai.nn.loss.cross_entropy_2p5d
:members:
colossalai.nn.loss.cross\_entropy\_3d
=====================================
.. automodule:: colossalai.nn.loss.cross_entropy_3d
:members:
colossalai.nn.loss
==================
.. automodule:: colossalai.nn.loss
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.loss.base_loss
colossalai.nn.loss.cross_entropy_1d
colossalai.nn.loss.cross_entropy_2d
colossalai.nn.loss.cross_entropy_2p5d
colossalai.nn.loss.cross_entropy_3d
colossalai.nn.lr\_scheduler.cosine
==================================
.. automodule:: colossalai.nn.lr_scheduler.cosine
:members:
colossalai.nn.lr\_scheduler.delayed
===================================
.. automodule:: colossalai.nn.lr_scheduler.delayed
: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