ipex-llm/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
Junwei Deng 6a7ab76a8e Nano: add a patch_torch/patch_tensorflow API for friendly usage (#5946)
* add a prototype code

* add patch_nano

* add pep8 changes

* add cv ppl

* fix bug

* add unpatch

* add updates

* update comment

* add changes

* add doc

* update and refactor

* update test

* fix doc

* fix according to cmt

* fix style

* fix doc

* fix bug

* fix style

* fix bug

* fix code

* add split patching api

* fix doc

* patch api to each framework

* fix pep8

* fix style

* fix style and cmt
2022-10-19 14:29:24 +08:00

34 lines
806 B
ReStructuredText

Nano PyTorch API
==================
bigdl.nano.pytorch.Trainer
---------------------------
.. autoclass:: bigdl.nano.pytorch.Trainer
:members:
: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
---------------------------
.. autoclass:: bigdl.nano.pytorch.TorchNano
:members:
:undoc-members:
:exclude-members: run
Patch API
---------------------------
.. automodule:: bigdl.nano.pytorch.dispatcher
:members:
:undoc-members:
:show-inheritance: