reqst

dingguanglei · Oct 26, 2018 · 16472ea · 16472ea
1 parent 402dc28
commit 16472ea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 103 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -22,7 +22,12 @@
                 'scipy.interpolate',
                 'torch','torchvision',
                 'tensorboardX',
-                'tensorboard'
+                'tensorboard',
+                'torch.optim',
+                'torch.nn',
+                'torch.utils.data',
+                'torch.autograd',
+                'torchvision.transforms'
                 ]
 for mod_name in MOCK_MODULES:
     sys.modules[mod_name] = mock.Mock()

diff --git a/jdit_train.py b/jdit_train.py