fix ImageFolder init bug with load_in_mem

Author: IShengFang

datasets.py

@@ -135,8 +135,8 @@ def __init__(self, root, transform=None, target_transform=None,
       print('Loading all images into memory...')
       self.data, self.labels = [], []
       for index in tqdm(range(len(self.imgs))):
-        path, target = self.transform(imgs[index][0]), imgs[index][1]
-        self.data.append(self.loader(path))
+        path, target = imgs[index][0], imgs[index][1]
+        self.data.append(self.transform(self.loader(path)))
         self.labels.append(target)