Skip to content

Commit 4b4795f

Browse files
committed
bugfix for lsun
1 parent 17cebdd commit 4b4795f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchvision/datasets/lsun.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ def __getitem__(self, index):
115115
if self.target_transform is not None:
116116
target = self.target_transform(target)
117117

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

120121
def __len__(self):
121122
return self.length

0 commit comments

Comments
 (0)