pytorch学习
Summer_crown
这个作者很懒,什么都没留下…
展开
-
ubuntu 下出现 cuDNN Error: CUDNN_STATUS_OUT_INITIALIZED错误
问题描述:换了一个模型跑代码,就出现了标题的问题 问题解决:可以看到出错之前是在loss.backward这个地方出错的,所以回去检查模型最后的输出outputs和标签labels获得的loss,发现在计算loss的时候用的是不带softmax的BCELoss() 函数,模型最后的输出也没有另外添加softmax()函数。所以出错了。 解决方案:1. 将outputs先输入softmax函数在进行criterion 2. 用自带softmax的损失函数BCEWithLogitsLoss作为criterio.原创 2021-07-12 10:08:53 · 232 阅读 · 0 评论 -
**Pytorch中的model.train()和model.eval()**
Pytorch中的model.train()和model.eval() - model.train() 相当于训练train阶段的一个初始化 设置Batch Normalization 和Droupout为真True Batch Normalization: 设置输入到神经网络的每一个batch具有相同的格式 Droupout:防止过拟合 - model.eval() 相当于测试test阶段的一个初始化 设置Batch Normalization 和Droupout为假FALSE 训练阶段都把参数训练好了,原创 2021-04-16 11:24:33 · 105 阅读 · 0 评论 -
**损失函数nll_loss**
损失函数nll_loss https://www.gentlecp.com/articles/874.html ↑↑↑有详细说明↑↑↑ nll_loss 函数接收两个tensor第一个是模型的output,第二个是label target,output中每一行与一个标签中每一列的元素对应,根据target的取值找出output行中对应位置元素,求和取平均值。 转载请注明:求索 » 【pytorch笔记】损失函数nll_loss ...转载 2021-04-16 10:52:58 · 401 阅读 · 0 评论