简单的tensorboard学习(pytorch)

2022-08-27 14:35:19

tensorboard是一个可视化工具,我在pytorch中使用,首先要先装这个包,利用下面代码:

pipinstall tensorboard

在利用测试代码看看效果:

from torch.utils.tensorboardimport SummaryWriter

writer= SummaryWriter("logs")# writer.add_image()# y = x**2for iinrange(100):
    writer.add_scalar("y = x**2", i**2, i)

writer.close()

然后在有logs文件夹下,按住shift+右键,点击“在此处打开Powershell窗口”,输入cmd,再输入代码激活环境

conda activate pytorch1.5

再输入代码,打开tensorboard

tensorboard --logdir=logs --port=6007

这时打开谷歌浏览器,输入

http://localhost:6007/

这样就可以看到我们画的图了

在这里插入图片描述

但我认为这个画图没有matplotlib好用,所以建议大家多利用matplotlib画图或者看参数变化。

  • 作者:爱听许嵩歌
  • 原文链接:https://blog.csdn.net/weixin_45092662/article/details/106330908
    更新时间:2022-08-27 14:35:19