每天五分钟深度学习pytorch：可视化神经网络训练损失函数图像
本文重点
我们希望训练的时候损失函数是降低的，一种好的办法就是通过画图的方式来将其可视化，这样效果比较好，其实这个很简单，只需要在训练过程中将训练损失和测试损失存储下来，然后使用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完成手写字体识别的损失图像是什么样的：
代码为：
原文地址：https://blog.csdn.net/huanfeng_AI/article/details/144302437
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！