pytorch中tensorboard的使用

pytorch中tensorboard的使用

首先安装tensorboard，在pytorch终端中执行如下命令

conda activate pytroch

pip3 install tensorboard

ide中写入如下代码：

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

# writer.add_image()

# y = x

for i in range(100):

writer.add_scalar("y = x", i, i)

writer.close()

执行

生成log文件

在pytorch终端中执行如下命令

tensorboard --logdir=logs

生产链接，打开即可看到对于函数图形