简单的tensorboard学习(pytorch)

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

pip install tensorboard

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

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

# writer.add_image()
# y = x**2
for i in range(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/

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

简单的tensorboard学习(pytorch)_第1张图片

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

你可能感兴趣的:(编程debug问题)