pytorch随机种子无法复现

随机种子设置:

random.seed(1)
np.random.seed(1)
torch.manual_seed(1)
cudnn.deterministic = True
torch.cuda.manual_seed_all(1)
cudnn.benchmark = False
os.environ['PYTHONHASHSEED'] = str(1)

以上设置了仍然可能会无法复现,如果你得代码加载数据的时候使用了numworkers,你需要确保你几份代码的numworkers的个数一样,不一样会直接导致结果不同。若个数相同还是会出现部分无法复现的情况…所以对复现很有需求建议numworkers设成0但是会影响训练速度…

猜你喜欢

转载自blog.csdn.net/qq_37668436/article/details/121044978
今日推荐