pytorch绘制训练过程中的曲线图

因为在训练的过程中想要输出一下accuracy和loss曲线,因此找到了一篇比较好的博客,原文链接:https://blog.csdn.net/tequilaro/article/details/81841748
但是过程中出现了问题,在尝试了一天之后终于解决了~

PyTorch

主要思想就是首先要定义两个数组，然后将每次迭代时模型的准确率和损失率传入到数组中，这里可以利用拼接函数进行拼接。然后在利用绘制图像的函数将其绘制。
代码如下:
首先定义两个数组:


#我这里迭代了300次，所以x的取值范围为(0，300)，然后再将每次相对应的准确率以及损失率附在x上

但是在绘制的过程中遇见了一个问题,报错显示x与y的维度不一样,尝试了各种方法,甚至将y也改成了range(0,5).(在尝试过程中我先将epoch设置成的5,这样会节省很多时间的哦~)虽然知道这个方法不对,但还是想看到输出结果.
好了,最大的问题就是缩进问题,真的是太坑了~
一定要先将所有的数值全部放在数组里之后再画,不然维数永远不可能相等~
缩进!缩进!缩进!!