TensorBoard的安装及使用
安装
pip install tensorboard
输入上述命令,回车等待安装即可
使用
使用该函数生成日志文件
from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter("logs")
# writer.add_image()
for i in range(100):
writer.add_scalar("y=x",i,i)
writer.close()
运行完上面的代码之后,可以看到在改项目下面生成了一个log文件夹
查看日志文件:
tensorboard --logdir=logs
默认使用的是6006端口
可以手动指定使用的端口:
tensorboard --logdir=logs --port=6009
在浏览器中输入网址或者点击网址自动打开浏览器该网址:
http://localhost:6009/
这里需要注意一下,有的同学可能出现下图所示的问题:
意思是没有写入任何事件文件!!!
可能是您的terminal所在的路径不对,一定要将目录进入到logs文件夹的上一层
比如我这里是在gan项目下面建立了test_code然后运行刚刚写入的事件的代码,它生成的logs文件夹默认跟py文件在同一层目录下,因为项目根目录 是gan,所以terminal打开默认路径也是gan,此时就要cd test_code
才可以运行tensorboard --logdir=logs --port=6009
该命令。