06cb1971c7
* Small fixes to (un)patch_tensorflow api doc and make its import path in api doc show as the recommended one * Add api doc for nano_bf16 decorator * Move api doc for bigdl.nano.tf.keras.InferenceOptimizer out of bigdl.nano.tf.keras to make it more clear * Fix python styles * Fix path in Nano PyTorch API docs * Add api doc for bigdl.nano.pytorch.patching.patch_encryption * Add a note box for bigdl.nano.pytorch.patching.patch_encryption api doc * Fix Python style again * Fix path in Nano HPO API doc and other small fixes
42 lines
849 B
ReStructuredText
42 lines
849 B
ReStructuredText
Nano HPO API
|
|
==================
|
|
|
|
Search Space
|
|
---------------------------
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Categorical
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Real
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Int
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Bool
|
|
|
|
|
|
HPO for Tensorflow
|
|
---------------------------
|
|
|
|
bigdl.nano.automl.tf.keras.Model
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. autoclass:: bigdl.nano.automl.tf.keras.Model
|
|
:members: search, search_summary
|
|
|
|
|
|
bigdl.nano.automl.tf.keras.Sequential
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. autoclass:: bigdl.nano.automl.tf.keras.Sequential
|
|
:members: search, search_summary
|
|
|
|
|
|
HPO for PyTorch
|
|
---------------------------
|
|
|
|
bigdl.nano.pytorch.Trainer
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. autoclass:: bigdl.nano.pytorch.Trainer
|
|
:members: search, search_summary
|
|
:undoc-members:
|
|
|