Hi,
I’m trying to use the function:
loss = criterion(outputs, labels)
where the inputs are:
labels=
tensor([[2]])
outputs=
tensor([[274.3095, 842.6060, -52.4284]], grad_fn=<AddmmBackward>)
but I get the error:
multi-target not supported at /Users/distiller/project/conda/conda-bld/pytorch_1570710797334/work/aten/src/THNN/generic/ClassNLLCriterion.c:22
Any help
Thanks