同一种子生成的随机数相同,用于下次复现实验结果。
random.seed(seed + worker_id)
np.random.seed(seed + worker_id)
torch.manual_seed(seed + worker_id)
torch.cuda.manual_seed(seed + worker_id)
torch.cuda.manual_seed_all(seed + worker_id)
同一种子生成的随机数相同,用于下次复现实验结果。
random.seed(seed + worker_id)
np.random.seed(seed + worker_id)
torch.manual_seed(seed + worker_id)
torch.cuda.manual_seed(seed + worker_id)
torch.cuda.manual_seed_all(seed + worker_id)