pytorch中tensorboardX可视化网络使用踩坑

pytorch中要使用tensorboard需要以下环境:
1.python 3.6+

2.Pytorch 0.4.0+

3.tensorboardX: pip install tensorboardX、pip install tensorflow

windows下要实现命令行操作tensorboard需要提前配置好环境变量,如果使用anaconda一般需要在系统变量的path中添 加C:\Users\XXX\Anaconda3\Scripts.

打开图形化界面需要在命令行键入tensorboard --logdir 目录
默认情况下在runs同级目录下使用命令行:tensorboard --logdir runs
当然也可以按ctrl+B对SummaryWriter中的log_dir=None的参数值进行修改,相关数据会保存在指定目录下。

注意: tensorboard需要图形化界面支持,要想在docker中使用必须在创建docker时使用支持图形化的命令。目前实验中只能采取将相关训练完成的文件夹复制之后在windows环境下操作,**调用的是本地解释器!**使用最新版本的tensorflow和tensorboardX避免版本出错。

1. scalar
可视化tensor数据,显示epoch,loss等训练信息,以下是通用样例:

import torch
from tensorboardX import SummaryWriter

writer = SummaryWriter()
x =
y =
for epoch in range(100):
	writer.add_histogram
  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
好的,我可以帮你解答这个问题! 首先,为了使用tensorboardx可视化Pytorch模型,你需要先安装tensorboardX库。你可以通过以下命令来安装: ``` pip install tensorboardX ``` 安装完成后,你需要在你的Pytorch代码导入tensorboardX: ```python from tensorboardX import SummaryWriter ``` 接下来,在你的代码创建一个`SummaryWriter`对象,它将负责记录你的模型的数据和可视化: ```python writer = SummaryWriter() ``` 现在你可以在代码使用`writer`对象来记录任何你想要记录的数据。例如,你可以记录损失函数的值: ```python writer.add_scalar('Loss', loss_value, global_step) ``` 其`loss_value`是损失函数的值,`global_step`是你的训练步数。 你也可以记录模型的权重和梯度: ```python writer.add_histogram('conv1/weights', conv1.weight, global_step) writer.add_histogram('conv1/grads', conv1.weight.grad, global_step) ``` 这将记录名为`conv1/weights`和`conv1/grads`的直方图,它们分别显示了`conv1`层的权重和梯度。 最后,在你的代码结束,不要忘记关闭`SummaryWriter`对象: ```python writer.close() ``` 现在你可以在终端输入以下命令来启动tensorboard: ``` tensorboard --logdir=/path/to/logs ``` 其`/path/to/logs`是你保存日志文件的路径。然后在你的浏览器访问`http://localhost:6006`,你将能够看到Pytorch模型的可视化结果。 希望这能够帮助你!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值