reqst

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

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -20,15 +20,15 @@
 MOCK_MODULES = ['numpy', 'scipy',
                 'matplotlib', 'matplotlib.pyplot',
                 'scipy.interpolate',
-                'torch','torchvision',
+                'torch',
                 'tensorboardX',
                 'tensorboard',
                 'torch.optim',
                 'torch.nn',
                 'torch.utils.data',
                 'torch.autograd',
-                'torchvision.transforms'
-                'torchvision.utils'
+                # 'torchvision.transforms','torchvision', 'torchvision.utils'
+
                 ]
 for mod_name in MOCK_MODULES:
     sys.modules[mod_name] = mock.Mock()

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