Expose VisionDataSet class so it can be modified (#876)

pytorch · Apr 29, 2019 · 3afcf3c · 3afcf3c
1 parent 0c36735
commit 3afcf3c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchvision/datasets/__init__.py b/torchvision/datasets/__init__.py
@@ -17,6 +17,7 @@
 from .caltech import Caltech101, Caltech256
 from .celeba import CelebA
 from .sbd import SBDataset
+from .vision import VisionDataset
 
 __all__ = ('LSUN', 'LSUNClass',
            'ImageFolder', 'DatasetFolder', 'FakeData',