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

Migrated project

parent 2ebaefc5
colossalai.nn.lr\_scheduler.linear
==================================
.. automodule:: colossalai.nn.lr_scheduler.linear
:members:
colossalai.nn.lr\_scheduler.multistep
=====================================
.. automodule:: colossalai.nn.lr_scheduler.multistep
:members:
colossalai.nn.lr\_scheduler.onecycle
====================================
.. automodule:: colossalai.nn.lr_scheduler.onecycle
:members:
colossalai.nn.lr\_scheduler.poly
================================
.. automodule:: colossalai.nn.lr_scheduler.poly
:members:
colossalai.nn.lr\_scheduler
===========================
.. automodule:: colossalai.nn.lr_scheduler
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.lr_scheduler.cosine
colossalai.nn.lr_scheduler.delayed
colossalai.nn.lr_scheduler.linear
colossalai.nn.lr_scheduler.multistep
colossalai.nn.lr_scheduler.onecycle
colossalai.nn.lr_scheduler.poly
colossalai.nn.lr_scheduler.torch
colossalai.nn.lr\_scheduler.torch
=================================
.. automodule:: colossalai.nn.lr_scheduler.torch
:members:
colossalai.nn.model.base\_model
===============================
.. automodule:: colossalai.nn.model.base_model
:members:
colossalai.nn.model.bert.bert
=============================
.. automodule:: colossalai.nn.model.bert.bert
:members:
colossalai.nn.model.bert
========================
.. automodule:: colossalai.nn.model.bert
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.model.bert.bert
colossalai.nn.model
===================
.. automodule:: colossalai.nn.model
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.model.bert
colossalai.nn.model.vanilla_resnet
colossalai.nn.model.vision_transformer
.. toctree::
:maxdepth: 2
colossalai.nn.model.base_model
colossalai.nn.model.vanilla\_resnet.resnet
==========================================
.. automodule:: colossalai.nn.model.vanilla_resnet.resnet
:members:
colossalai.nn.model.vanilla\_resnet
===================================
.. automodule:: colossalai.nn.model.vanilla_resnet
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.model.vanilla_resnet.resnet
colossalai.nn.model.vision\_transformer
=======================================
.. automodule:: colossalai.nn.model.vision_transformer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.model.vision_transformer.vision_transformer
colossalai.nn.model.vision\_transformer.vision\_transformer
===========================================================
.. automodule:: colossalai.nn.model.vision_transformer.vision_transformer
:members:
colossalai.nn.multi\_tensor\_apply.multi\_tensor\_apply
=======================================================
.. automodule:: colossalai.nn.multi_tensor_apply.multi_tensor_apply
:members:
colossalai.nn.multi\_tensor\_apply
==================================
.. automodule:: colossalai.nn.multi_tensor_apply
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.multi_tensor_apply.multi_tensor_apply
colossalai.nn.optimizer.fp16\_optimizer
=======================================
.. automodule:: colossalai.nn.optimizer.fp16_optimizer
:members:
colossalai.nn.optimizer.fused\_adam
===================================
.. automodule:: colossalai.nn.optimizer.fused_adam
:members:
colossalai.nn.optimizer.fused\_lamb
===================================
.. automodule:: colossalai.nn.optimizer.fused_lamb
:members:
colossalai.nn.optimizer.fused\_sgd
==================================
.. automodule:: colossalai.nn.optimizer.fused_sgd
: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