654baef1a3
* add base optimizer for future InferenceOptimizer keras * fix type * add type change * change doc
35 lines
863 B
ReStructuredText
35 lines
863 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: ALL_INFERENCE_ACCELERATION_METHOD
|
|
:inherited-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:
|