Skip to content

Commit c933729

Browse files
iacolippofmassa
authored andcommitted
Fix bug in LSUN dataset for test split. (#435)
Both conditional branches in `__init__` were executed, since `classes` for test becomes a list and `if` was used instead of `elif`.
1 parent f124509 commit c933729

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/datasets/lsun.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def __init__(self, root, classes='train',
8585
classes = [classes]
8686
else:
8787
classes = [c + '_' + classes for c in categories]
88-
if type(classes) == list:
88+
elif type(classes) == list:
8989
for c in classes:
9090
c_short = c.split('_')
9191
c_short.pop(len(c_short) - 1)

0 commit comments

Comments
 (0)