Add Nano PyTorch API Doc (#4730)

* add trainer api doc

* add index and fix trainer

* fix doc

* reset

* fix conflict

* fix import

docs/readthedocs/requirements-doc.txt

@@ -6,9 +6,10 @@ sphinxemoji
 click
 tensorflow==1.15.2
 bigdl==0.12.0
+cloudpickle
 ray[tune]==1.9.2
 ray==1.9.2
-torch==1.7.1
+torch==1.9.0
 Pygments==2.3.1
 setuptools==41.0.1
 docutils==0.14
@@ -27,6 +28,6 @@ pmdarima
 sphinx_markdown_tables
 numpy==1.21.2
 xgboost
-torchmetrics
+torchmetrics==0.7.2
 pickle5==0.0.12
 protobuf~=3.19.0

docs/readthedocs/source/conf.py

@@ -18,6 +18,8 @@ import glob
 import shutil
 import urllib
 
+autodoc_mock_imports = ["openvino", "pytorch_lightning"]
+
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 #sys.path.insert(0, '.')
 sys.path.insert(0, os.path.abspath('.'))
@@ -26,6 +28,7 @@ sys.path.insert(0, os.path.abspath("../../../python/chronos/src/"))
 sys.path.insert(0, os.path.abspath("../../../python/dllib/src/"))
 sys.path.insert(0, os.path.abspath("../../../python/orca/src/"))
 sys.path.insert(0, os.path.abspath("../../../python/serving/src/"))
+sys.path.insert(0, os.path.abspath("../../../python/nano/src/"))
 
 
 

docs/readthedocs/source/doc/PythonAPI/Nano/index.rst

@@ -0,0 +1,7 @@
+Nano API
+==================
+
+.. toctree::
+    :maxdepth: 2
+
+    pytorch.rst

docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst

@@ -0,0 +1,10 @@
+Nano PyTorch API
+==================
+
+bigdl.nano.pytorch
+---------------------------
+
+.. 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

docs/readthedocs/source/index.rst

@@ -125,6 +125,7 @@ BigDL Documentation
    doc/PythonAPI/Orca/orca.rst
    doc/PythonAPI/Friesian/feature.rst
    doc/PythonAPI/Chronos/index.rst
+   doc/PythonAPI/Nano/index.rst
 
 .. toctree::
    :maxdepth: 1