We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f124509 commit c933729Copy full SHA for c933729
torchvision/datasets/lsun.py
@@ -85,7 +85,7 @@ def __init__(self, root, classes='train',
85
classes = [classes]
86
else:
87
classes = [c + '_' + classes for c in categories]
88
- if type(classes) == list:
+ elif type(classes) == list:
89
for c in classes:
90
c_short = c.split('_')
91
c_short.pop(len(c_short) - 1)
0 commit comments