fix doc build

Author: rejuvyesh

mkdocs.yml

@@ -63,6 +63,7 @@ plugins:
       enable_inventory: true
       handlers:
         python:
+          paths: [src]
           import:
             - https://docs.python.org/3/objects.inv
             - https://pytorch.org/docs/stable/objects.inv