Skip to content

GitLab

Commit 3afcf3cd authored by bbowles-personal's avatar bbowles-personal Committed by Francisco Massa
Browse files

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

parent 0c36735d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
torchvision/datasets/__init__.py
View file @ 3afcf3cd
...@@ -17,6 +17,7 @@ from .imagenet import ImageNet ...@@ -17,6 +17,7 @@ from .imagenet import ImageNet
from .caltech import Caltech101, Caltech256 from .caltech import Caltech101, Caltech256
from .celeba import CelebA from .celeba import CelebA
from .sbd import SBDataset from .sbd import SBDataset
from .vision import VisionDataset
__all__ = ('LSUN', 'LSUNClass', __all__ = ('LSUN', 'LSUNClass',
'ImageFolder', 'DatasetFolder', 'FakeData', 'ImageFolder', 'DatasetFolder', 'FakeData',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment