[pytorch] pytorch 优雅画图

tensorboard太丑了，用plt

pytorch学习之如何画损失函数曲线图_骊山道童的博客-CSDN博客_损失函数曲线

import matplotlib.pyplot as plt
plt.switch_backend('Agg')
#1.训练时先新建个列表，然后将loss值调用列表的append方法存入列表中
#2.例如列表train_recon_loss，Discriminator_loss...，然后将列表名替换train_recon_loss，Discriminator，利用plot即可画出曲线
#3.最后将画的图保存成图片，imgpath为自定义的图片保存路径。
# plt.figure(num = 2, figsize=(640,480))
plt.figure()
plt.plot(plt_loss, 'b', label='loss')
plt.ylabel('loss')
plt.xlabel('iter_num')
plt.legend()
plt.savefig(os.path.join(self.args.model_dir, "loss.jpg"))