为了方便后续使用模型，可以将训练过程写成一个函数，向该函数传入网络模型、损失函数、优化器等必要对象后，在MNIST数据集上进行训练并打印日志观察过程，代码如下：
调用train(1)训练一轮的结果如下，可以看出loss值不断下降。其中，1表示训练1轮数据集。真正训练时，则可以训练多轮，比如调用train(20)表示训练20轮。
