diff --git a/docs/source/fastNLP.core.callbacks.rst b/docs/source/fastNLP.core.callbacks.rst index 89d85f52..d0f3d210 100644 --- a/docs/source/fastNLP.core.callbacks.rst +++ b/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 diff --git a/docs/source/fastNLP.core.collators.padders.rst b/docs/source/fastNLP.core.collators.padders.rst index 6f40becb..0c50dd4c 100644 --- a/docs/source/fastNLP.core.collators.padders.rst +++ b/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 diff --git a/docs/source/fastNLP.core.dataloaders.rst b/docs/source/fastNLP.core.dataloaders.rst index e8c6b799..db53dbe0 100644 --- a/docs/source/fastNLP.core.dataloaders.rst +++ b/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 diff --git a/docs/source/fastNLP.core.dataloaders.torch_dataloader.rst b/docs/source/fastNLP.core.dataloaders.torch_dataloader.rst index c9acca23..a3aeb1bf 100644 --- a/docs/source/fastNLP.core.dataloaders.torch_dataloader.rst +++ b/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 diff --git a/docs/source/fastNLP.core.drivers.rst b/docs/source/fastNLP.core.drivers.rst index bb168c76..30652fec 100644 --- a/docs/source/fastNLP.core.drivers.rst +++ b/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 diff --git a/docs/source/fastNLP.core.drivers.torch_driver.rst b/docs/source/fastNLP.core.drivers.torch_driver.rst index 9a0109a2..c9080a86 100644 --- a/docs/source/fastNLP.core.drivers.torch_driver.rst +++ b/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 diff --git a/docs/source/fastNLP.core.metrics.backend.rst b/docs/source/fastNLP.core.metrics.backend.rst index 5a8cf4ad..4466a54a 100644 --- a/docs/source/fastNLP.core.metrics.backend.rst +++ b/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 diff --git a/docs/source/fastNLP.core.utils.rst b/docs/source/fastNLP.core.utils.rst index 2d682010..9bf76a23 100644 --- a/docs/source/fastNLP.core.utils.rst +++ b/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 diff --git a/docs/source/fastNLP.io.loader.rst b/docs/source/fastNLP.io.loader.rst index bd91b795..20be532a 100644 --- a/docs/source/fastNLP.io.loader.rst +++ b/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 diff --git a/docs/source/fastNLP.io.pipe.rst b/docs/source/fastNLP.io.pipe.rst index 9ad7e539..53a62918 100644 --- a/docs/source/fastNLP.io.pipe.rst +++ b/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 diff --git a/docs/source/fastNLP.io.rst b/docs/source/fastNLP.io.rst index 5f025bba..7e1a5a67 100644 --- a/docs/source/fastNLP.io.rst +++ b/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 diff --git a/docs/source/fastNLP.modules.rst b/docs/source/fastNLP.modules.rst index fa1d95de..b686105d 100644 --- a/docs/source/fastNLP.modules.rst +++ b/docs/source/fastNLP.modules.rst @@ -13,3 +13,4 @@ Subpackages :maxdepth: 4 fastNLP.modules.mix_modules + fastNLP.modules.torch diff --git a/docs/source/fastNLP.rst b/docs/source/fastNLP.rst index 89c8e058..55776c22 100644 --- a/docs/source/fastNLP.rst +++ b/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 \ No newline at end of file diff --git a/docs/source/fastNLP.transformers.rst b/docs/source/fastNLP.transformers.rst index c681acca..023da63d 100644 --- a/docs/source/fastNLP.transformers.rst +++ b/docs/source/fastNLP.transformers.rst @@ -1,6 +1,5 @@ fastNLP.transformers package -======================= - +============================ .. automodule:: fastNLP.transformers :members: :undoc-members: diff --git a/docs/source/fastNLP.transformers.torch.rst b/docs/source/fastNLP.transformers.torch.rst index ab6ef210..29080d73 100644 --- a/docs/source/fastNLP.transformers.torch.rst +++ b/docs/source/fastNLP.transformers.torch.rst @@ -1,5 +1,5 @@ fastNLP.transformers.torch package -======================= +============================= .. automodule:: fastNLP.transformers.torch :members: