linux下使用tensorboadX可视化pytorch训练结果
2021/10/31 7:10:40
本文主要是介绍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
这篇关于linux下使用tensorboadX可视化pytorch训练结果的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-04Linux系统上离线升级SSH服务的具体操作步骤-icode9专业技术文章分享
- 2024-06-0600-macOS和Linux安装和管理多个Python版本
- 2024-03-30[译]漫画SELinux概念
- 2024-03-29linux 移动文件
- 2024-03-28linux .so file
- 2024-03-28Linux 磁盘管理
- 2024-03-28Linux学习笔记(十三)磁盘管理(一):磁盘分区
- 2024-03-26linux 创建 文件
- 2024-03-25使用SecureCRT对Linux vim进行颜色设置
- 2024-03-202019-2020-12 20199317 《Linux内核原理与分析》 第十二周作业