pytorch使用tensorboardX进行可视化需要进行的操作

本文详细介绍了如何在 PyTorch 1.6 中使用 TensorBoardX 进行模型训练过程的可视化,包括安装步骤、配置方法及浏览器展示效果。文中还提供了使用不同浏览器查看日志文件的方法以及调整端口号的技巧。
部署运行你感兴趣的模型镜像

pytorch1.6使用tensorboardX进行可视化,安装记录如下:
在这里插入图片描述

  1. 使用Anaconda Prompt输入绝对路径,参考链接https://blog.youkuaiyun.com/qq_40317204/article/details/107443694。 老版本的tensorboard --logdir与文件地址之间可能需要用等号连接,并且文件地址没有双引号。
    在这里插入图片描述
    复制后使用谷歌浏览器打开,亲测360极速浏览器也可以打开。
  2. 使用pycharm terminal ,tensorboard --logdir “./相对路径” (绝对路径亲测也行)在这里插入图片描述
    打开链接,如下图所示:
    在这里插入图片描述
  3. 还可以修改输出的端口,在引号后面加上 --port=604在这里插入图片描述

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

好的,我可以帮你解答这个问题! 首先,为了使用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模型的可视化结果。 希望这能够帮助你!
评论 2
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值