您当前的位置: 首页 >  ar

耐心的小黑

暂无认证

  • 0浏览

    0关注

    323博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

pytorch学习笔记九:通过torch.utils.tensorboard在pytorch中使用tensorboard

耐心的小黑 发布时间:2020-04-09 14:58:21 ,浏览量:0

参考文章:详解PyTorch项目使用TensorboardX进行训练可视化

这篇文章虽然说的是TensorboardX,但其实只要我们把TensorboardX直接换成torch.utils.tensorboard就可以了,至少经过我的测试是没有问题的,如果有问题欢迎大家指出来!!!

一、环境准备:
  • 我的环境是最新的pytorch1.4,然后使用pip install tensorflow在pytorch中安装tensorflow,此时默认安装的是tf2.1,同时它还会默认安装我们需要的tensorboard
  • 注意,有朋友可能有疑问:既然我们是直接使用的torch.utils.tensorboard这个pytorch自己的API,那么我们为什么还要安装tensorflow呢?其实一开始我也有这样的疑问,后来看了一下这个API的实现方式之后发现,其实是因为它内部的代码实现就是用的tensorflow,所以我们必须安装tensorflow
二、初步了解tensorboard应该如何使用

首先,需要创建一个 SummaryWriter 的示例:

from torch
关注
打赏
1640088279
查看更多评论
立即登录/注册

微信扫码登录

0.0432s