From 3b28ac2e3d3c3ebfff48000502be24cf315e0cf2 Mon Sep 17 00:00:00 2001 From: Mark Zakharov Date: Tue, 6 Aug 2019 22:49:18 +0300 Subject: [PATCH] Switched shuffles for train and test loaders --- utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index 51fb0a9..f86c19f 100644 --- 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