running on cpu Loading file... 986410 Generating testset... 0%| | 0/10000 [00:00) tensor([9, 8, 4]) 0%| | 1/3333 [00:00<17:48, 3.12it/s]tensor([[[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]], [[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]], [[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]]], grad_fn=) tensor([[[9, 9, 9, 9, 9, 9, 9, 9, 9, 9]], [[2, 2, 2, 2, 2, 2, 2, 2, 2, 2]], [[9, 9, 9, 9, 9, 9, 9, 9, 9, 9]]]) 0%| | 1/3333 [00:00<18:14, 3.04it/s] Traceback (most recent call last): File "pytorch_ai.py", line 147, in loss = loss_function(output.view(-1, 10), label) File "/home/clemens/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__ result = self.forward(*input, **kwargs) File "/home/clemens/.local/lib/python3.7/site-packages/torch/nn/modules/loss.py", line 916, in forward ignore_index=self.ignore_index, reduction=self.reduction) File "/home/clemens/.local/lib/python3.7/site-packages/torch/nn/functional.py", line 2009, in cross_entropy return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction) File "/home/clemens/.local/lib/python3.7/site-packages/torch/nn/functional.py", line 1838, in nll_loss ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index) RuntimeError: multi-target not supported at /pytorch/aten/src/THNN/generic/ClassNLLCriterion.c:22