RuntimeError: multi-target not supported at C:\w\1\s\tmp_conda_3.7_104508\conda\conda-bld\pytorch_1572950778684\work\aten\src\THNN/generic/ClassNLLCriterion.c:22
本文这里是交叉熵函数要求第二个变量维度应该为N,而本人这里是NX1,因此reshape下就可以了
(N为变量的个数哈)
这里由于本人是循环训练,有时候个数不是固定的
因此取得是NX1(label_total的维度)中的N:label_total.shape[0]
label_total = label_total.reshape(label_total.shape[0])
loss_d = criterion_t(preds_d, label_total)