Pytorch模型训练技巧－－seed

最新推荐文章于 2024-10-21 20:45:48 发布

哆啦A梦！！！

最新推荐文章于 2024-10-21 20:45:48 发布

阅读量1.8k

点赞数 1

文章标签： pytorch 深度学习神经网络

本文链接：https://blog.csdn.net/weixin_44737266/article/details/118542312

版权

模型训练中的随机种子设置

作用：固定住深度模型训练的过程，使得每次从头开始训练模型初始化方式和数据读取方式保持一致
使用：

seed = 1
np.random.seed(seed)
torch.manual_seed(seed)
torch.backends.cudnn.deterministic = True   # 保证每次卷积算法返回结果一样
torch.backends.cudnn.benchmark = False　　　# 保证每次cudnn使用的都是同一种算法，而不是自行选择最优算法