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