一。toch.nn.CELoss
最近在使用该loss的时候遇到了如下三个问题。
-
One-hot问题。
RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1587428398394/work/aten/src/THCUNN/generic/ClassNLLCriterion.cu:18
参考此篇博客.这是因为CEloss期望的真值输入不能是one-hot形式的,打印一下原本的预测值和真值维度、部分数据如下:pred shape: torch.Size([128, 10])
target shape: torch.Size([128, 10])
pred: tensor([ -1.8647, 11.9629, -8.9586, -4.3062, -2.7667, 2.8557, -1.5226,-12.8641, 3.2789, 0.8572], device=‘cuda:0’, grad_fn=)
target: tensor([0., 0., 0., 0., 1., 0., 0., 0., 0., 0.], device=‘cuda:0’)
CELoss在内部运算过程中会自动进行one-hot编码,所以,这里target中对应维度应该是【128, 1】, 具体到上述打印中的例子, 应该就是单维张量tensor(4, device='cuda:0')
。
2. 数据类型问题
RuntimeError: Expected object of scalar type Long but got sca