-
安装tensorboard:
Cmd命令:pip install tensorboardX
-
在训练中导入:
from tensorboardX import SummaryWriter
-
在程序中调用:
writer = SummaryWriter(‘log’)
保存要做图的所有信息,在目录下建立一个log文件夹存放。
4. 绘制loss曲线:
writer.add_scalar('Train/Loss', loss.data[0], niter)
在训练过程中,依次写入:图像名称;loss数值;n_interation。
5. 绘制准确度曲线:
writer.add_scalar('Test/Accu', correct/total, niter)
在测试的循环中,写入预测的准确度即可。
6. 加载图像文件:
Cmd命令:cd 到log文件夹同级目录,然后tensorboard –logdir=log
,回车
产生一个网页地址,copy到谷歌浏览器打开。(不知为何无法打开,解决办法,直接谷歌浏览器输入:http://localhost:6006,问题解决。)