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

最近被模型的可复现问题弄得不堪其扰,现在勉强算是解决了,记录一下 首先是最常规的随机种子设置 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设备设置随机种子,但 官方文档 中已...

Continue Reading »
分享到: