pytorch训练dnn模型

pytorch官方github：https://github.com/pytorch/examples/blob/main/mnist/main.py

pytorch文档：https://pytorch.org/docs/stable/generated/torch.nn.NLLLoss.html#torch.nn.NLLLoss