reqst

dingguanglei · Oct 26, 2018 · cedd8ac · cedd8ac
1 parent c167ce3
commit cedd8ac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -21,13 +21,15 @@
                 'matplotlib', 'matplotlib.pyplot',
                 'scipy.interpolate',
                 'torch',
-                'tensorboardX',
+                'tensorboardX','tensorboardX.SummaryWriter'
                 'tensorboard',
                 'torch.optim',
-                'torch.nn',
+                'torch.nn','torch.nn',
                 'torch.utils.data',
-                'torch.autograd',
-                # 'torchvision.transforms','torchvision', 'torchvision.utils'
+                'torch.autograd','torch.autograd.Variable'
+                'torchvision.transforms','torchvision', 'torchvision.utils',
+                'torchvision.utils.make_grid','torchvision.datasets',
+
 
                 ]
 for mod_name in MOCK_MODULES:

diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,6 @@ pip
 mock
 future
 #numpy
-torchvision
 pyyaml
 setuptools
 six