def METHOD_NAME(self) -> str:
    size = "256" if self.small else "large"
    if self.split.startswith("train"):
        dir = f"data_{size}_{self.variant}"
    else:
        dir = f"{self.split}_{size}"
    return path.join(self.root, dir)