From edcab051a9f98ebdaf813543205978945c655fd0 Mon Sep 17 00:00:00 2001 From: Ruonan Wang <105281011+rnwang04@users.noreply.github.com> Date: Thu, 15 Sep 2022 18:43:47 +0800 Subject: [PATCH] Nano : add API for InferenceOptimizer and deprecated some Trainer APIs (#5768) * add api and deprecated * add license * add deprecated * update comment for api display * modify comment * update based on comment --- docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst index 52a87cd2278..b7d6530561e 100644 --- a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst +++ b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst @@ -9,6 +9,14 @@ bigdl.nano.pytorch.Trainer :undoc-members: :exclude-members: accelerator_connector, checkpoint_connector, reload_dataloaders_every_n_epochs, limit_val_batches, logger, logger_connector, state +bigdl.nano.pytorch.InferenceOptimizer +--------------------------- + +.. autoclass:: bigdl.nano.pytorch.InferenceOptimizer + :members: + :undoc-members: + :exclude-members: + bigdl.nano.pytorch.TorchNano ---------------------------