diff --git a/torchvision/datasets/places365.py b/torchvision/datasets/places365.py index ba6292a1e11..9880266b0fe 100644 --- a/torchvision/datasets/places365.py +++ b/torchvision/datasets/places365.py @@ -113,7 +113,7 @@ def process(line: str) -> Tuple[str, int]: images = [process(line) for line in fh] _, targets = zip(*images) - return images, targets + return images, list(targets) def download_devkit(self) -> None: file, md5 = self._DEVKIT_META["challenge" if self.split == "train-challenge" else "standard"]