diff --git a/torchvision/datasets/lsun.py b/torchvision/datasets/lsun.py index d437596b74e..75b284b597f 100644 --- a/torchvision/datasets/lsun.py +++ b/torchvision/datasets/lsun.py @@ -83,7 +83,7 @@ def __init__( self.dbs = [] for c in self.classes: self.dbs.append(LSUNClass( - root=root + '/' + c + '_lmdb', + root=os.path.join(root, f"{c}_lmdb"), transform=transform)) self.indices = []