Skip to content

Commit 2c277d6

Browse files
committed
removing dataset postfix
1 parent aaa218d commit 2c277d6

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

torchvision/datasets/__init__.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
from .lsun import LSUNDataset, LSUNClassDataset
2-
from .folder import ImageFolderDataset
3-
from .coco import CocoCaptionsDataset, CocoDetectionDataset
1+
from .lsun import LSUN, LSUNClass
2+
from .folder import ImageFolder
3+
from .coco import CocoCaptions, CocoDetection
44

5-
__all__ = ('LSUNDataset', 'LSUNClassDataset',
6-
'ImageFolderDataset',
7-
'CocoCaptionsDataset', 'CocoDetectionDataset')
5+
__all__ = ('LSUN', 'LSUNClass',
6+
'ImageFolder',
7+
'CocoCaptions', 'CocoDetection')

0 commit comments

Comments
 (0)