跳到主要内容
NEUSNCP / 知识空间

博客

发现有价值的观点,记录自己的思考。

写博客
szp

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设备设置随机种子,但 官方文档 中已...