Seed for the DataLoader

vision
1

Hi,all
I want to get the same data from DataLoader when I use DataLoader twice.
I found this https://github.com/pytorch/pytorch/issues/7068
and follow this reference but fail.
I also do

`worker_init_fn=np.random.seed(0)` in DataLoader option.

torch.manual_seed(0)
torch.cuda.manual_seed(0)
np.random.seed(0)
random.seed(0)
torch.backends.cudnn.deterministic = True

source_test_loader = torch.utils.data.DataLoader(source_val, batch_size=params.batch_size, shuffle=True, num_workers=params.num_workers, drop_last=True, pin_memory=False)
2

this should be a callable, not a number. You can read the doc, which is pretty clear on how to use worker_init_fn.