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

3afcf3cd · bbowles-personal · Francisco Massa · 0c36735d · 3afcf3cd
Commit 3afcf3cd authored Apr 29, 2019 by bbowles-personal Committed by Francisco Massa Apr 29, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

torchvision/datasets/__init__.py torchvision/datasets/__init__.py +1 -0

No files found.
--- a/torchvision/datasets/__init__.py
+++ b/torchvision/datasets/__init__.py
@@ -17,6 +17,7 @@ from .imagenet import ImageNet
 from .caltech import Caltech101, Caltech256
 from .celeba import CelebA
 from .sbd import SBDataset
+from .vision import VisionDataset

 __all__ = ('LSUN', 'LSUNClass',
           'ImageFolder', 'DatasetFolder', 'FakeData',