torch nll

正确格式：

data：3,3,3,2

label：3,3,2

2.data:3:2

label:2

3.data:2,1,6

label:2,6

Traceback (most recent call last):
torch.Size([3, 3, 3, 2]) torch.Size([3, 2])
File "F:/project/chedaoxian/Ultra-Fast-bar-Detection/utils/loss.py", line 91, in <module>
out_size, target.size()))
ValueError: Expected target size (3, 3, 2), got torch.Size([3, 2])

data维度 [2,6]

label维度[2]，这样才可以，增加维度就报错。

如果data是[2,1,6]

label就需要是[2,6]

  import torchimport torch.nn as nnimport torch.nn.functional as Fdata = torch.randn(2,6)target = torch.tensor([1,2])print('data:', data.size(),target.size())entropy_out = F.cross_entropy(data, target)print('entropy_out:', entropy_out, target.size())log_soft = F.log_softmax(data, dim=1)print('log_soft:', log_soft, '\n')nll_out = F.nll_loss(log_soft, target)print('nll_out:', nll_out)

torch nll_loss相关推荐

nll_loss 和 cross_entropy
nll_loss torhc.nn.functional中给出的关于nll_loss的一个样例 target: 数值 <= C-1, 一般可以看作一个一维列表,存放的是真实类别编号先经过sof ...
【Pytorch】model.train()和model.eval()用法和区别，以及model.eval()和torch.no_grad()的区别
model.train() 启用 Batch Normalization 和 Dropout 如果模型中有BN层(Batch Normalization)和Dropout,需要在训练时添加model. ...
深入理解Pytorch负对数似然函数（torch.nn.NLLLoss)和交叉熵损失函数（torch.nn.CrossEntropyLoss)
在看Pytorch的交叉熵损失函数torch.nn.CrossEntropyLoss官方文档介绍中,给出的表达式如下.不免有点疑惑为何交叉熵损失的表达式是这个样子的 loss ⁡ ( y , clas ...
torch使用tensorboard简明备忘录
tensorboard是让torch可以使用tensorflow的web可视化工具,之前叫tensorboardX. 至于其他的介绍以及为什么需要,可自行百度. 简单的完整代码 12345678910 ...
【pytorch笔记】损失函数nll_loss
文章目录使用场景函数理解例子演示总结使用场景在用pytorch做训练.测试时经常要用到损失函数计算输出与目标结果的差距,例如下面的代码: # 训练 for batch_idx, (data ...
交叉熵损失函数和NLL_loss
目录 1.交叉熵损失函数的计算过程: 2.交叉熵损失函数的公式: 3.NLL loss(负对数似然函数损失) 4.NLL_Loss和交叉熵损失函数关系 1.交叉熵损失函数的计算过程: 交叉熵损失函数经 ...
DL - softmax、log_softmax、nll_loss、cross_entropy_loss
文章目录 softmax log_softmax nll_loss cross_entropy_loss 关于 log_softmax 输出使用 log_softmax 计算 softmax 的对数值 ...
NLL_LOSS与CROSS_ENTROPY_LOSS详解
NLL_LOSS, CROSS_ENTROPY_LOSS详解常用损失函数 nll cross_entropy 一段代码 transformer中的一段经典代码常用损失函数 nll 官方文档 tor ...
pytorch中torch.optim的介绍
pytorch中torch.optim的介绍这是torch自带的一个优化器,里面自带了求导,更新等操作.开门见山直接讲怎么使用: 常用的引入: import torch.optim as optim ...

torch nll_loss

torch nll_loss相关推荐

最新文章

热门文章