参考文章:详解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
首先,需要创建一个 SummaryWriter 的示例:
from torch