Pytorch 02 绘制acc loss 曲线

1、空白列表

acc = [ ]

loss = [ ]

 

对于epoch来说，每epoch记录一次acc，和loss 

2、设置y轴信息 ：像列表中添加信息

steps = len(dataloader)

acc.append(epoch_acc / steps )

loss.append(epoch_loss / steps)

 

3、设定x轴信息

x1 = range(0:steps)

x2 = range(0:steps)

 

4、绘制图像

plt.subplot(211)

plt.plot(x1,acc,"-0")

plt.title('acc vs epoch')

plt.ylabel("acc")

 

plt.subplot(212)

plt.plot(x2,loss,"-0")

plt.title('loss vs epoch')

plt.ylabel("loss")

 

plt.show()

plt.savefig("acc.png")