Commit c9337292 authored by Iacopo Poli's avatar Iacopo Poli Committed by Francisco Massa
Browse files

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`.
parent f124509a
...@@ -85,7 +85,7 @@ class LSUN(data.Dataset): ...@@ -85,7 +85,7 @@ class LSUN(data.Dataset):
classes = [classes] classes = [classes]
else: else:
classes = [c + '_' + classes for c in categories] classes = [c + '_' + classes for c in categories]
if type(classes) == list: elif type(classes) == list:
for c in classes: for c in classes:
c_short = c.split('_') c_short = c.split('_')
c_short.pop(len(c_short) - 1) c_short.pop(len(c_short) - 1)
......
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