rst

docs/source/fastNLP.core.callbacks.rst

@@ -31,5 +31,6 @@ Submodules
    fastNLP.core.callbacks.lr_scheduler_callback
    fastNLP.core.callbacks.more_evaluate_callback
    fastNLP.core.callbacks.progress_callback
+   fastNLP.core.callbacks.timer_callback
    fastNLP.core.callbacks.topk_saver
    fastNLP.core.callbacks.utils

docs/source/fastNLP.core.collators.padders.rst

@@ -16,6 +16,7 @@ Submodules
    fastNLP.core.collators.padders.get_padder
    fastNLP.core.collators.padders.jittor_padder
    fastNLP.core.collators.padders.numpy_padder
+   fastNLP.core.collators.padders.oneflow_padder
    fastNLP.core.collators.padders.padder
    fastNLP.core.collators.padders.paddle_padder
    fastNLP.core.collators.padders.raw_padder

docs/source/fastNLP.core.dataloaders.rst

@@ -13,6 +13,7 @@ Subpackages
    :maxdepth: 4
 
    fastNLP.core.dataloaders.jittor_dataloader
+   fastNLP.core.dataloaders.oneflow_dataloader
    fastNLP.core.dataloaders.paddle_dataloader
    fastNLP.core.dataloaders.torch_dataloader
 
@@ -22,6 +23,5 @@ Submodules
 .. toctree::
    :maxdepth: 4
 
-   fastNLP.core.dataloaders.mix_dataloader
    fastNLP.core.dataloaders.prepare_dataloader
    fastNLP.core.dataloaders.utils

docs/source/fastNLP.core.dataloaders.torch_dataloader.rst

@@ -13,3 +13,4 @@ Submodules
    :maxdepth: 4
 
    fastNLP.core.dataloaders.torch_dataloader.fdl
+   fastNLP.core.dataloaders.torch_dataloader.mix_dataloader

docs/source/fastNLP.core.drivers.rst

@@ -13,6 +13,7 @@ Subpackages
    :maxdepth: 4
 
    fastNLP.core.drivers.jittor_driver
+   fastNLP.core.drivers.oneflow_driver
    fastNLP.core.drivers.paddle_driver
    fastNLP.core.drivers.torch_driver
 

docs/source/fastNLP.core.drivers.torch_driver.rst

@@ -13,9 +13,11 @@ Submodules
    :maxdepth: 4
 
    fastNLP.core.drivers.torch_driver.ddp
+   fastNLP.core.drivers.torch_driver.deepspeed
    fastNLP.core.drivers.torch_driver.dist_utils
-   fastNLP.core.drivers.torch_driver.fairscale_sharded
+   fastNLP.core.drivers.torch_driver.fairscale
    fastNLP.core.drivers.torch_driver.initialize_torch_driver
    fastNLP.core.drivers.torch_driver.single_device
    fastNLP.core.drivers.torch_driver.torch_driver
+   fastNLP.core.drivers.torch_driver.torch_fsdp
    fastNLP.core.drivers.torch_driver.utils

docs/source/fastNLP.core.metrics.backend.rst

@@ -13,6 +13,7 @@ Subpackages
    :maxdepth: 4
 
    fastNLP.core.metrics.backend.jittor_backend
+   fastNLP.core.metrics.backend.oneflow_backend
    fastNLP.core.metrics.backend.paddle_backend
    fastNLP.core.metrics.backend.torch_backend
 

docs/source/fastNLP.core.utils.rst

@@ -16,7 +16,10 @@ Submodules
    fastNLP.core.utils.dummy_class
    fastNLP.core.utils.exceptions
    fastNLP.core.utils.jittor_utils
+   fastNLP.core.utils.oneflow_utils
    fastNLP.core.utils.paddle_utils
    fastNLP.core.utils.rich_progress
+   fastNLP.core.utils.seq_len_to_mask
    fastNLP.core.utils.torch_utils
+   fastNLP.core.utils.tqdm_progress
    fastNLP.core.utils.utils

docs/source/fastNLP.io.loader.rst

@@ -14,7 +14,6 @@ Submodules
 
    fastNLP.io.loader.classification
    fastNLP.io.loader.conll
-   fastNLP.io.loader.coreference
    fastNLP.io.loader.csv
    fastNLP.io.loader.cws
    fastNLP.io.loader.json

docs/source/fastNLP.io.pipe.rst

@@ -15,7 +15,6 @@ Submodules
    fastNLP.io.pipe.classification
    fastNLP.io.pipe.conll
    fastNLP.io.pipe.construct_graph
-   fastNLP.io.pipe.coreference
    fastNLP.io.pipe.cws
    fastNLP.io.pipe.matching
    fastNLP.io.pipe.pipe

docs/source/fastNLP.io.rst

@@ -25,5 +25,4 @@ Submodules
    fastNLP.io.embed_loader
    fastNLP.io.file_reader
    fastNLP.io.file_utils
-   fastNLP.io.model_io
    fastNLP.io.utils

docs/source/fastNLP.modules.rst

@@ -13,3 +13,4 @@ Subpackages
    :maxdepth: 4
 
    fastNLP.modules.mix_modules
+   fastNLP.modules.torch

docs/source/fastNLP.rst

@@ -13,6 +13,10 @@ Subpackages
    :maxdepth: 4
 
    fastNLP.core
+   fastNLP.embeddings
    fastNLP.envs
    fastNLP.io
+   fastNLP.models
    fastNLP.modules
+   .. 以下为手动添加
+   fastNLP.transformers

docs/source/fastNLP.transformers.rst

@@ -1,6 +1,5 @@
 fastNLP.transformers package
-=======================
-
+============================
 .. automodule:: fastNLP.transformers
    :members:
    :undoc-members:

docs/source/fastNLP.transformers.torch.rst

@@ -1,5 +1,5 @@
 fastNLP.transformers.torch package
-=======================
+=============================
 
 .. automodule:: fastNLP.transformers.torch
    :members: