Switched shuffles for train and test loaders

anordertoreclaim · Aug 6, 2019 · 3b28ac2 · 3b28ac2
1 parent 048ff8f
commit 3b28ac2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils.py b/utils.py
@@ -69,7 +69,7 @@ def get_loaders(dataset_name, batch_size, color_levels, train_root, test_root):
     except KeyError:
         raise AttributeError("Unsupported dataset")
 
-    train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=False, pin_memory=True, drop_last=True)
-    test_loader = DataLoader(test_dataset, batch_size=batch_size, shuffle=True, pin_memory=True, drop_last=True)
+    train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True, pin_memory=True, drop_last=True)
+    test_loader = DataLoader(test_dataset, batch_size=batch_size, shuffle=False, pin_memory=True, drop_last=True)
 
     return train_loader, test_loader, h, w