🎈 原创

挺好用的一个小工具
先简单记录一下，不常用容易忘

导入

from torch.utils.tensorboard import SummaryWriter

类初始化

writer = SummaryWriter("logs") # 一般用writer这个名字

def __init__(self,log_dir=None,comment="",purge_step=None,max_queue=10,flush_secs=120,filename_suffix="",)

# 参数log_dir: 写入的文件夹，默认./run  
# 其他的不常用  
# comment= 加在文件名最后的内容

# 大致用法
writer = SummaryWriter() # 初始化
writer.add_scalar() # 各种绘图
# ...
# ...
writer.close() # 关闭

添加函数

writer.add_scalar()

参数：
tag: 每次都要改，不然会将多个log文件的值渲染到一张图中
    解决方法：
        - 将log文件夹中删掉对应
        - 指定一个新文件夹
scalar_value: 相当于y值
global_step: 相当于x值

writer.add_image()

类型：torch.tensor / numpy
tensor的颜色通道应该放在第0维上（官方规定），与numpy不同
这个函数默认(3,H,W)，如果不是，需要指定dataformats参数
dataformats: e.g. 'CHW' 'HWC' 'HW'

除此之外的各种用法还在探索中…

打开TensorBoard

TensorBoard不是直接在python文件中使用，需要在Terminal中，进入install了tensorboard的环境使用

# 指定log文件夹（一定）
tensorboard --logdir=logs
# 可以指定自己的端口
tensorboard --logdir=logs --port=6007