13 Tensorboard简单使用

Tensorboard 简单使用

参考链接：

https://pytorch.org/docs/stable/tensorboard.html

https://zhuanlan.zhihu.com/p/103630393

最近发现tensorboard可以直接省去画图的时间，挺好用，记录下：

1、安装

pip install tensorboard

使用 vscode 的话，IDE 会自动提醒你安装依赖的环境。

2、使用

# 导入tensorboard
from torch.utils.tensorboard import SummaryWriter

# 实例化
writer = SummaryWriter('')	# 可设置log_dir
# writer = SummaryWriter()	自动生成log_dir：runs/
# writer = SummaryWriter("my_experiment") 生成指定的log_dir: my_experiment/
# writer = SummaryWriter(comment="LR_0.1_BATCH_16") 生成指定的log_dir: runs/LR_0.1_BATCH_16/

# 将数值写入文件
writer.add_scalar(
	tag, 				# 图的标签
	scalar_value, 		# 图的y轴
	global_step=None, 	# 图的x轴
	walltime=None)		# 省略即可

未完，待续……