Skip to content

Commit

Permalink
Nano : add API for InferenceOptimizer and deprecated some Trainer APIs (
Browse files Browse the repository at this point in the history
#5768)

* add api and deprecated

* add license

* add deprecated

* update comment for api display

* modify comment

* update based on comment
  • Loading branch information
rnwang04 authored Sep 15, 2022
1 parent d07e862 commit edcab05
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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
---------------------------

Expand Down

0 comments on commit edcab05

Please sign in to comment.