Commit efa4322d authored by cormoran's avatar cormoran Committed by Soumith Chintala
Browse files

fix inconsistent behavior for ~ expression (#850)

* fix inconsistent behavior for ~ expression

* remove duplicated code
parent bbaa1b0d
...@@ -82,10 +82,10 @@ class DatasetFolder(VisionDataset): ...@@ -82,10 +82,10 @@ class DatasetFolder(VisionDataset):
super(DatasetFolder, self).__init__(root) super(DatasetFolder, self).__init__(root)
self.transform = transform self.transform = transform
self.target_transform = target_transform self.target_transform = target_transform
classes, class_to_idx = self._find_classes(root) classes, class_to_idx = self._find_classes(self.root)
samples = make_dataset(root, class_to_idx, extensions) samples = make_dataset(self.root, class_to_idx, extensions)
if len(samples) == 0: if len(samples) == 0:
raise (RuntimeError("Found 0 files in subfolders of: " + root + "\n" raise (RuntimeError("Found 0 files in subfolders of: " + self.root + "\n"
"Supported extensions are: " + ",".join(extensions))) "Supported extensions are: " + ",".join(extensions)))
self.loader = loader self.loader = loader
......
...@@ -31,7 +31,7 @@ class Omniglot(VisionDataset): ...@@ -31,7 +31,7 @@ class Omniglot(VisionDataset):
def __init__(self, root, background=True, def __init__(self, root, background=True,
transform=None, target_transform=None, transform=None, target_transform=None,
download=False): download=False):
super(Omniglot, self).__init__(join(os.path.expanduser(root), self.folder)) super(Omniglot, self).__init__(join(root, self.folder))
self.transform = transform self.transform = transform
self.target_transform = target_transform self.target_transform = target_transform
self.background = background self.background = background
......
...@@ -39,7 +39,7 @@ class SEMEION(VisionDataset): ...@@ -39,7 +39,7 @@ class SEMEION(VisionDataset):
self.data = [] self.data = []
self.labels = [] self.labels = []
fp = os.path.join(root, self.filename) fp = os.path.join(self.root, self.filename)
data = np.loadtxt(fp) data = np.loadtxt(fp)
# convert value to 8 bit unsigned integer # convert value to 8 bit unsigned integer
# color (white #255) the pixels # color (white #255) the pixels
......
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