网站首页
编程语言
数据库
系统相关
其他分享
编程问答
losses
2024-12-06
每天五分钟深度学习pytorch:可视化神经网络训练损失函数图像
本文重点我们希望训练的时候损失函数是降低的,一种好的办法就是通过画图的方式来将其可视化,这样效果比较好,其实这个很简单,只需要在训练过程中将训练损失和测试损失存储下来,然后使用matplotlib来绘图。绘图核心代码train_losses.append(train_loss/len(train_data))#将训练损