- update conf.py

inferno-pytorch · Sep 1, 2017 · f42ea65 · f42ea65
1 parent da24ac0
commit f42ea65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -29,7 +29,7 @@ def __getattr__(cls, name):
                 'argparse',
                 'pandas',
                 'torch',
-                'torch.nn', 'torch.nn.init',
+                'torch.nn', 'torch.nn.init', 'torch.nn.functional', 'torch.nn.parallel',
                 'torch.multiprocessing', 'torch.autograd',
                 'torch.optim', 'torch.sparse', 'torch.cuda']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)