pytorch中提高模型可复现性的几个设置

2022-04-19 10:42 szp

最近被模型的可复现问题弄得不堪其扰，现在勉强算是解决了，记录一下 首先是最常规的随机种子设置 python中随机种子 random.seed(seed) numpy中随机种子 np.random.seed(seed) torch中随机种子 torch.manual_seed(seed) 部分博客中还记录了给GPU设备设置随机种子的设置，torch.cuda.manual_seed给当前GPU设备设置随机种子，torch.cuda.manual_seed_all给所有GPU设备设置随机种子，但 官方文档 中已...