Fix preprocessing bug

yijiuzai · Sep 21, 2017 · d2f37d2 · d2f37d2
1 parent 9274797
commit d2f37d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data.py b/data.py
@@ -91,8 +91,8 @@ def get_batch(self, dataset_name, augment=False):
 
             x_support_set = np.array(x_augmented_support_set)
             x_target = np.array(x_augmented_target_set)
-            x_support_set = self.preprocess_batch(x_support_set)
-            x_target = self.preprocess_batch(x_target)
+        x_support_set = self.preprocess_batch(x_support_set)
+        x_target = self.preprocess_batch(x_target)
 
         return x_support_set, y_support_set, x_target, y_target