什么是deterministic training(确定性训练),pytorch如何实现

deterministic training(确定性训练)是一种训练机器学习模型的方式,其主要特点是确保在相同的初始条件下,每次训练都会得到相同的结果。换句话说,确定性训练旨在消除随机性,使得模型的训练过程是可重复和可预测的。

deterministic training可以确保模型的结果可复现,从而有利于不同研究者互相对比参考实验结果。

想要实现deterministic training,首先需要知道模型随机性的来源,有如下几个方面:初始化权重、数据加载的顺序、dropout、优化器的随机性(如随机梯度下降中的随机梯度选择)、CUDA的随机性等。

想要用pytorch实现deterministic training就需要固定上面提到的随机性因素,另外,超参数也要保障每次都相同。


# 设置随机数种子  
torch.manual_seed(0)  
torch.cuda.manual_seed_all(0)  # GPU种子也要设置  
torch.backends.cudnn.deterministic = True  # 设置CUDNN为deterministic  
torch.backends.cudnn.benchmark = False  # 关闭CUDNN的自动调优,确保每次运行都一致  
  

data = torch.randn(100, 10)  
dataset = TensorDataset(data, target)  
loader = DataLoader(dataset, batch_size=10, shuffle=False)  # shuffle设为False以保证数据顺序  

以上就可以做到deterministic training,但是由于不同硬件带来的些微结果差距是始终无法消除的,不过一般来说也影响不大。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝海渔夫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值