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")