Tensorboard使用

官方文档

tensorboard是可视化工具

1、安装：
用Anaconda，输入以下命令

conda install tensorboard

或者，
在vscode或者pycharm的终端输入以下命令

pip install tensorboard

2、使用：

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")  #将数据以特定的格式存储到logs的文件夹上


for i in range(100):
    
# writer.add_image()
    writer.add_scalar("y = x ", i , i)
#scalar_value is y , global_step is x
writer.close()

HOW TO RUN?

在vscode终端输入：

tensorboard --logdir=logs #logs是保存的路径

这样，就可以给出网址了

如果想要修改端口，输入以下命令：

tensorboard --logdir=logs --port=6007

如果发现，结果图不对，建议把logs（这里是我定义的文件夹）下的文件全部删掉，重新跑，或者新建一个文件夹