Skip to content

Commit 3771b47

Browse files
committed
Fix VisionDataset class inheritance omission
1 parent acccf86 commit 3771b47

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/datasets/gtsrb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def __init__(
3535
super().__init__(root, transform=transform, target_transform=target_transform)
3636

3737
self._split = verify_str_arg(split, "split", ("train", "test"))
38-
self._base_folder = pathlib.Path(root) / "gtsrb"
38+
self._base_folder = pathlib.Path(self.root) / "gtsrb"
3939
self._target_folder = (
4040
self._base_folder / "GTSRB" / ("Training" if self._split == "train" else "Final_Test/Images")
4141
)

0 commit comments

Comments
 (0)