diff --git a/torchvision/datasets/gtsrb.py b/torchvision/datasets/gtsrb.py index e6b60116c40..5db85d260bb 100644 --- a/torchvision/datasets/gtsrb.py +++ b/torchvision/datasets/gtsrb.py @@ -35,7 +35,7 @@ def __init__( super().__init__(root, transform=transform, target_transform=target_transform) self._split = verify_str_arg(split, "split", ("train", "test")) - self._base_folder = pathlib.Path(root) / "gtsrb" + self._base_folder = pathlib.Path(self.root) / "gtsrb" self._target_folder = ( self._base_folder / "GTSRB" / ("Training" if self._split == "train" else "Final_Test/Images") )