/home/ps/.local/lib/python3.8/site-packages/torch/nn/_reduction.py:42: UserWarning: size_average and reduce args will be deprecated,
test_loss += F.nll_loss(output, target, size_average=False).item()
修正代码:
test_loss += F.nll_loss(output, target, reduction='sum').item()