Bug fix

MadryLab · Nov 1, 2019 · 1ba7881 · 1ba7881
1 parent 5d4b186
commit 1ba7881
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/robustness/datasets.py b/robustness/datasets.py
@@ -97,8 +97,8 @@ def get_model(self, arch, pretrained):
         raise NotImplementedError
 
     def make_loaders(self, workers, batch_size, data_aug=True, subset=None, 
-                     subset_start=0, subset_type='rand', val_batch_size=None,
-                     only_val=False):
+                 subset_start=0, subset_type='rand', val_batch_size=None,
+                 only_val=False, shuffle_train=True, shuffle_val=True):
         '''
         Args:
             workers (int) : number of workers for data fetching (*required*).
@@ -146,7 +146,9 @@ def make_loaders(self, workers, batch_size, data_aug=True, subset=None,
                                     subset=subset,
                                     subset_start=subset_start,
                                     subset_type=subset_type,
-                                    only_val=only_val)
+                                    only_val=only_val,
+                                    shuffle_train=shuffle_train,
+                                    shuffle_val=shuffle_val)
 
 class ImageNet(DataSet):
     '''

diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@
   # Versions should comply with PEP440.  For a discussion on single-sourcing
   # the version across setup.py and the project code, see
   # https://packaging.python.org/en/latest/single_source_version.html
-  version='1.1',
+  version='1.1.post1',
 
   description='Tools for Robustness',
   long_description=long_description,