您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:pytorch预测之解决多次预测结果不一致问题

51自学网 2021-10-30 22:35:55
  python
这篇教程pytorch预测之解决多次预测结果不一致问题写得很实用,希望能帮到您。

为什么多次预测结果不一致

1、检查是否在每次预测前使用

model.eval()

或者是

with torch.no_grad():   for ...

推荐下面的方法,上面的的方法计算梯度,但是并不反向传播,下面的方法既不计算梯度,也不反向传播,速度更快。

2、检查是否取消了所有的dropout

3、设置随机种子

def setup_seed(seed):    np.random.seed(seed)    random.seed(seed)    torch.manual_seed(seed) #cpu    torch.cuda.manual_seed_all(seed)  #并行gpu    torch.backends.cudnn.deterministic = True  #cpu/gpu结果一致    torch.backends.cudnn.benchmark = True   #训练集变化不大时使训练加速

4、保证实例化模型前要将is_training置为false;这两行代码顺序不能颠倒

以上为个人经验,希望能给大家一个参考,也希望大家多多支持51zixue.net。


python异常的传递知识点总结
matplotlib画混淆矩阵与正确率曲线的实例代码
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。