From 793969a3cad329294492d7167b992c13ea3448b6 Mon Sep 17 00:00:00 2001 From: wogong Date: Wed, 6 Nov 2019 22:20:35 +0800 Subject: [PATCH] using random seed. --- datasets/gtsrb.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/datasets/gtsrb.py b/datasets/gtsrb.py index b5003c4..ee02d86 100644 --- a/datasets/gtsrb.py +++ b/datasets/gtsrb.py @@ -3,7 +3,7 @@ import os import torch -from torchvision import datasets, transforms +from torchvision import transforms import torch.utils.data as data from torch.utils.data.sampler import SubsetRandomSampler import numpy as np @@ -31,7 +31,6 @@ class GTSRB(data.Dataset): def __len__(self): return self.n_data - def get_gtsrb(dataset_root, batch_size, train): """Get GTSRB datasets loader.""" shuffle_dataset = True @@ -51,7 +50,8 @@ def get_gtsrb(dataset_root, batch_size, train): dataset_size = len(gtsrb_dataset) indices = list(range(dataset_size)) if shuffle_dataset: - np.random.seed(random_seed) + #np.random.seed(random_seed) + np.random.seed() np.random.shuffle(indices) train_indices, val_indices = indices[:train_size], indices[train_size:]