reqst

dingguanglei · Oct 26, 2018 · e72860b · e72860b
1 parent cedd8ac
commit e72860b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -28,10 +28,14 @@
                 'torch.utils.data',
                 'torch.autograd','torch.autograd.Variable'
                 'torchvision.transforms','torchvision', 'torchvision.utils',
+                'torchvision.transforms.Resize','torchvision.transforms.ToTensor',
+                'torchvision.transforms.Normalize',
                 'torchvision.utils.make_grid','torchvision.datasets',
+                'torchvision.datasets.MNIST','torchvision.datasets.FashionMNIST',
+                'torchvision.datasets.CIFAR10','torchvision.datasets.LSUN',
+                ]
 
 
-                ]
 for mod_name in MOCK_MODULES:
     sys.modules[mod_name] = mock.Mock()
 import torch