糖尿病康复,内容丰富有趣,生活中的好帮手!
糖尿病康复 > pytorch在固定随机数种子后训练神经网络得到的结果仍然不一样

pytorch在固定随机数种子后训练神经网络得到的结果仍然不一样

时间:2023-05-25 18:48:09

相关推荐

pytorch在固定随机数种子后训练神经网络得到的结果仍然不一样

在训练一个分类器的时候最前面加入了以下代码固定随机数种子,但是每次重复实验得到的结果还是都不一样。

def seed_torch(seed=0):random.seed(seed)np.random.seed(seed)os.environ['PYTHONHASHSEED'] = str(seed)torch.manual_seed(seed)torch.cuda.manual_seed(seed)torch.backends.cudnn.deterministic = Truetorch.backends.cudnn.benchmark = False

最后发现是在主函数里面,定义模型之后加入了

cudnn.benchmark = True

这句话可以提高模型的运行效率,但是也是这个原因导致了每次的结果不一样

去掉之后完美解决结果无法复现的问题

如果觉得《pytorch在固定随机数种子后训练神经网络得到的结果仍然不一样》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。