pytorch学习之如何画损失函数曲线图
-
PS：本帖主要记录和解决如何画损失函数曲线的问题，希望可以帮到您。
作者介绍
张伟伟，男，西安工程大学电子信息学院，2019级硕士研究生，张宏伟人工智能课题组。 研究方向：机器视觉与人工智能。 电子邮件：zhangweiweicpp@163.com
-
.
内容简介
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(train_recon_loss,b,label = Recon_loss) plt.ylabel(Recon_loss) plt.xlabel(iter_num) plt.legend() plt.savefig(os.path.join(imgPath,"1_recon_loss.jpg")) plt.figure() plt.plot(train_loss_adv,r,label = Discriminator_loss) plt.ylabel(Discriminator_loss) plt.xlabel(iter_num) plt.legend() plt.savefig(os.path.join(imgPath,"2_train_loss_adv.jpg"))
上一篇：
JS实现多线程数据分片下载