tensorboard的简单使用

---

安装

pip install tensorboard

---

简单使用案例

更多语法查看官方文档https://www.tensorflow.org/tensorboard/get_started?hl=zh-cn

try:
	from tensorboard import SummaryWriter
except:
	from torch.utils.tensorboard import SummaryWriter
sw = SummaryWriter("/path/to/log_dir") # 不指定目录的话默认是当前目录下的runs(没有的话会自动创建)
sw.add_scalars("loss", {"train_loss":train_loss_avg, "test_loss":test_loss_avg}, epoch)
sw.add_scalars("score", {"score": score}, epoch)

---

启动tensorboard

1. 命令行启动

# 按需替换 /path/to/log_dir 与端口号，将输出的连接输入浏览器进行浏览
tensorboard --log_dir=/path/to/log_dir --port=8889 --bind_all

2. vscode 插件浏览 (python 插件里自带 tensorboard)

在 vscode 的命令栏(Ctrl + Shift + P)执行 Python: Launch TensorBoard 命令，根据提示选择要执行的目录即可，vscode 会自动打开一个 tensorboard 浏览页面。