Update docs

zuoxingdong · Sep 4, 2018 · 0cc5c1f · 0cc5c1f
1 parent 3d3b3b8
commit 0cc5c1f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -18,13 +18,7 @@
 
 # Exclude some dependencies for Read the Docs to compile with
 import mock
-MOCK_MODULES = [#'torch', 
-                #'torch.nn', 
-                #'torch.nn.functional',
-                #'torch.nn.utils',
-                #'torch.optim', 
-                #'torch.distributions', 
-                'seaborn', 
+MOCK_MODULES = ['seaborn', 
                 'matplotlib',
                 'matplotlib.pyplot',
                 '_tkinter']

diff --git a/lagom/core/networks/base_network.py b/lagom/core/networks/base_network.py
@@ -12,7 +12,7 @@ class BaseNetwork(nn.Module):
     
     - :meth:`make_params`
     - :meth:`init_params`
-    
+    - :meth:`forward`
     
     Example::