Update load_corrupted_data.py

xjtushujun · May 20, 2020 · bd1fd3e · bd1fd3e
1 parent 1fdb779
commit bd1fd3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/load_corrupted_data.py b/load_corrupted_data.py
@@ -76,7 +76,7 @@ def __init__(self, root='', train=True, meta=True, num_meta=1000,
         self.meta = meta
         self.corruption_prob = corruption_prob
         self.num_meta = num_meta
-
+        np.random.seed(seed)
         if download:
             self.download()
 
@@ -177,7 +177,7 @@ def __init__(self, root='', train=True, meta=True, num_meta=1000,
                 else:
                     assert False, "Invalid corruption type '{}' given. Must be in {'unif', 'flip', 'hierarchical'}".format(corruption_type)
 
-                np.random.seed(seed)
+
                 if corruption_type == 'clabels':
                     mean = [x / 255 for x in [125.3, 123.0, 113.9]]
                     std = [x / 255 for x in [63.0, 62.1, 66.7]]