2.1.5 训练网络

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

输出结果如下: