Skip to content

GitLab

Unverified Commit 63870514 authored by Sergii Dymchenko's avatar Sergii Dymchenko Committed by GitHub
Browse files

Simplify _check_integrity for cifar and stl10 (#6335)

parent 1a1d509c
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 8 deletions
torchvision/datasets/cifar.py
View file @ 63870514
...@@ -126,10 +126,8 @@ class CIFAR10(VisionDataset): ...@@ -126,10 +126,8 @@ class CIFAR10(VisionDataset):
return len(self.data) return len(self.data)
def _check_integrity(self) -> bool: def _check_integrity(self) -> bool:
root = self.root for filename, md5 in self.train_list + self.test_list:
for fentry in self.train_list + self.test_list: fpath = os.path.join(self.root, self.base_folder, filename)
filename, md5 = fentry[0], fentry[1]
fpath = os.path.join(root, self.base_folder, filename)
if not check_integrity(fpath, md5): if not check_integrity(fpath, md5):
return False return False
return True return True
......
torchvision/datasets/stl10.py
View file @ 63870514
...@@ -145,10 +145,8 @@ class STL10(VisionDataset): ...@@ -145,10 +145,8 @@ class STL10(VisionDataset):
return images, labels return images, labels
def _check_integrity(self) -> bool: def _check_integrity(self) -> bool:
root = self.root for filename, md5 in self.train_list + self.test_list:
for fentry in self.train_list + self.test_list: fpath = os.path.join(self.root, self.base_folder, filename)
filename, md5 = fentry[0], fentry[1]
fpath = os.path.join(root, self.base_folder, filename)
if not check_integrity(fpath, md5): if not check_integrity(fpath, md5):
return False return False
return True return True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment