Skip to content

Commit 41df18e

Browse files
pmeierNicolasHug
andauthored
Apply suggestions from code review
Co-authored-by: Nicolas Hug <[email protected]>
1 parent e647f45 commit 41df18e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/datasets/fer2013.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class FER2013(VisionDataset):
1616
1717
Args:
1818
root (string): Root directory of dataset where directory
19-
``caltech101`` exists or will be saved to if download is set to True.
19+
``root/fer2013`` exists.
2020
split (string, optional): The dataset split, supports ``"train"`` (default), or ``"test"``.
2121
transform (callable, optional): A function/transform that takes in an PIL image and returns a transformed
2222
version. E.g, ``transforms.RandomCrop``
@@ -63,7 +63,7 @@ def __getitem__(self, idx: int) -> Tuple[Any, Any]:
6363
return image, target
6464

6565
def _verify_integrity(self):
66-
base_folder = os.path.join(self.root, type(self).__name__.lower())
66+
base_folder = os.path.join(self.root, "fer2013")
6767
file_name, md5 = self._RESOURCES[self._split]
6868
file = os.path.join(base_folder, file_name)
6969
if not check_integrity(file, md5=md5):

0 commit comments

Comments
 (0)