pytorch中怎么保证每次运行，batch_size中选择的图片是一样的，程序具有可复现性

    def worker_init_fn(worker_id):
        np.random.seed(1 + worker_id)

    train_data_loader = DataLoader(train_dataset, batch_size=args.batch_size,
                                   shuffle=True, num_workers=args.num_workers, 
                                   pin_memory=True,
drop_last=True,worker_init_fn=worker_init_fn)

worker_init_fn控制每次选的图像是一样的