如何在pycharm上画loss函数图像

如何在pycharm上画loss函数图像

(即loss与迭代次数之间的关系图像)
1.代码:
在train函数下面加入:
writer.add_scalar(‘loss/train_loss’, losses.val, global_step=epoch)
2.完成训练后,在pycharm的Terminal处输入:
tensorboard --logdir="./runs"

注意
这里可能会提示tensorboard不是外部命令啊什么的,出现这个的原因是:
1)没有安装tensorboard(注:一般安装过Tensorflow的都安装过tensorboard)
2)没有把tensorboard的路径加入环境变量
在系统环境变量中加入下面路径即可。
在这里插入图片描述

成功运行后,点击下方显示的浏览器,即可看到loss函数图像。
如何在pycharm上画loss函数图像_第1张图片

如何在pycharm上画loss函数图像_第2张图片

你可能感兴趣的:(pycharm,机器学习)