系统相关
首页 > 系统相关> > linux下使用tensorboadX可视化pytorch训练结果

linux下使用tensorboadX可视化pytorch训练结果

作者:互联网

1、依赖

首先你需要安装torch、tensorflow、tensorboardX

2、基本代码

from tensorboardX import SummaryWriter
writer = SummaryWriter(logdir) # logdir是你自己的保存记录的目录
for i in range(epoch):
	loss = model(xxx)
	writer.add_scalar('train/loss', loss.item(), i)
writer.close()

3、使用tensorboadx

如果你有多个Python,需要指定你当前使用的python的tensorboard,比如:

python xxx/site-packages/tensorboard/main.py --logdir=/data02/gob/project/document_event_extraction/tensorboard --port=6006

我们在自己的python环境下找到site-packages/tensorboard/main.py所在的路径,然后输入python xxx/site-packages/tensorboard/main.py --logdir=xxx --port=6006指令即可。

4、可视化结果

我们在linux下难以输入localhost:6006来在浏览器查看结果,因此,可以在window下建立一个映射,使得我们可以在window下查看linux下tensorboard的结果,这里我们需要的工具是xshell,具体可以参考:
https://blog.csdn.net/weixin_42059276/article/details/106668679
如果尝试了还是不成功的话,就新建一个6006到6006的映射,然后在浏览器输入:127.0.0.1:6006

标签:6006,python,xxx,--,pytorch,logdir,tensorboard,linux,tensorboadX
来源: https://www.cnblogs.com/xiximayou/p/15487232.html