running on cpu Loading file... 986410 Generating testset... 0%| | 0/10000 [00:00 loss = F.nll_loss(output.view(1, 10), label[0]) RuntimeError: shape '[1, 10]' is invalid for input of size 1500