minor fix for python3

df75fa63 · Sean Robertson · Adam Paszke · 04fde085 · df75fa63
Commit df75fa63 authored Feb 23, 2017 by Sean Robertson Committed by Adam Paszke Feb 24, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/torchvision/datasets/coco.py
+++ b/torchvision/datasets/coco.py
@@ -8,7 +8,7 @@ class CocoCaptions(data.Dataset):
        from pycocotools.coco import COCO
        self.root = root
        self.coco = COCO(annFile)
-        self.ids = self.coco.imgs.keys()
+        self.ids = list(self.coco.imgs.keys())
        self.transform = transform
        self.target_transform = target_transform