2.1.5 训练网络

我们只需要在数据迭代器上循环传给网络和优化器的输入就可以，训练模型代码如下：

输出结果如下：