bugfix for lsun

4b4795f9 · soumith · 17cebddd · 4b4795f9
Commit 4b4795f9 authored Nov 16, 2016 by soumith
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

torchvision/datasets/lsun.py torchvision/datasets/lsun.py +2 -1

No files found.
--- a/torchvision/datasets/lsun.py
+++ b/torchvision/datasets/lsun.py
@@ -115,7 +115,8 @@ class LSUN(data.Dataset):
        if self.target_transform is not None:
            target = self.target_transform(target)

-        return db[index], target
+        img, _ = db[index]
+        return img, target

    def __len__(self):
        return self.length