update documentation

docs/colossalai/colossalai.nn.data.prefetcher.rst

@@ -1,5 +0,0 @@
-colossalai.nn.data.prefetcher
-=============================
-
-.. automodule:: colossalai.nn.data.prefetcher
-   :members:

docs/colossalai/colossalai.nn.data.rst

@@ -16,5 +16,3 @@ colossalai.nn.data
    colossalai.nn.data.base_dataset
    colossalai.nn.data.caltech101_dataset
    colossalai.nn.data.cifar10_dataset
-   colossalai.nn.data.prefetcher
-   colossalai.nn.data.wiki_dataset

docs/colossalai/colossalai.nn.data.wiki_dataset.rst

@@ -1,5 +0,0 @@
-colossalai.nn.data.wiki\_dataset
-================================
-
-.. automodule:: colossalai.nn.data.wiki_dataset
-   :members:

docs/colossalai/colossalai.nn.model.bert.bert.rst

@@ -1,5 +0,0 @@
-colossalai.nn.model.bert.bert
-=============================
-
-.. automodule:: colossalai.nn.model.bert.bert
-   :members:

docs/colossalai/colossalai.nn.model.bert.rst

@@ -1,11 +0,0 @@
-colossalai.nn.model.bert
-========================
-
-.. automodule:: colossalai.nn.model.bert
-   :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.nn.model.bert.bert

docs/colossalai/colossalai.nn.model.rst

@@ -7,7 +7,6 @@ colossalai.nn.model
 .. toctree::
    :maxdepth: 2
 
-   colossalai.nn.model.bert
    colossalai.nn.model.vanilla_resnet
    colossalai.nn.model.vision_transformer
 

docs/colossalai/colossalai.nn.optimizer.lars.rst

@@ -0,0 +1,5 @@
+colossalai.nn.optimizer.lars
+============================
+
+.. automodule:: colossalai.nn.optimizer.lars
+   :members:

docs/colossalai/colossalai.nn.optimizer.rst

@@ -13,6 +13,7 @@ colossalai.nn.optimizer
    colossalai.nn.optimizer.fused_lamb
    colossalai.nn.optimizer.fused_sgd
    colossalai.nn.optimizer.lamb
+   colossalai.nn.optimizer.lars
    colossalai.nn.optimizer.loss_scaler
    colossalai.nn.optimizer.zero_redundancy_optimizer_level_1
    colossalai.nn.optimizer.zero_redundancy_optimizer_level_2