Fix LSUN IndexError (#177)

83263d85 · Zhou Le · Soumith Chintala · cd8c0c3c · 83263d85
Commit 83263d85 authored May 28, 2017 by Zhou Le Committed by Soumith Chintala May 28, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/torchvision/datasets/lsun.py
+++ b/torchvision/datasets/lsun.py
@@ -114,7 +114,7 @@ class LSUN(data.Dataset):
            if index < ind:
                break
            target += 1
-            sub += ind
+            sub = ind

        db = self.dbs[target]
        index = index - sub