本文重点
我们希望训练的时候损失函数是降低的，一种好的办法就是通过画图的方式来将其可视化，这样效果比较好，其实这个很简单，只需要在训练过程中将训练损失和测试损失存储下来，然后使用matplotlib来绘图。
绘图核心代码
train_losses.append(train_loss/len(train_data))#将训练损失添加到训练损失列表 valid_losses.append(valid_loss/len(test_data))#将测试损失添加到测试损失列表
plt.plot(train_losses, label='train')#绘图
plt.plot(valid_losses, label='valid')
plt.xlabel('epoch')
plt.legend(loc='best')
plt.show()
实例
我们可视化一下CNN完成手写字体识别的损失图像是什么样的：
代码为：
