tensorboad使用方法

tensorboad

运行

tensorboad --logdir 日志目录

浏览器打开

http://localhost:6006/

相关用法

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter(log_dir=<日志目录，string类型>)

# writer.add_scalar,绘制单条曲线
writer.add_scalar(tag=<图像的名称-string类型>,
                  scalar_value=<值-y轴>,
                  global_step=<第几次迭代-x轴> )
# writer.add_scalars，绘制多条曲线
writer.add_scalars(main_tag=<图像的名称-string类型>,
                  tag_scalar_dict = {key1:val1,key2:val2...},
                  global_step=<第几次迭代-x轴>)

# writer.add_image，保存图
writer.add_image(tag=<图像的名称-string类型>, 
                 img_tensor=<图像（torch.Tensor， np.array，or string/blobname）>,
                 global_step=<第几次迭代-x轴>,
                 dataformats=<默认'HWC'，（通道C，高度H，宽度W），可修改>)
# close
writer.close()

now_time = datetime.now().strftime("%m_%d_%H_%M")