So sorry, seems I forgot it:
-> 1052         return torch._C._nn.nll_loss(input, target, weight, size_average, ignore_index, reduce)
1053     elif dim == 4:
1054         return torch._C._nn.nll_loss2d(input, target, weight, size_average, ignore_index, reduce)
RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1518244421288/work/torch/lib/THNN/generic/ClassNLLCriterion.c:22