pytorch中tensorboard的add_scalar与add_image

一、add_scalar

add_scalar(tag, scalar_value, global_step=None, )：将我们所需要的数据保存在文件里面供可视化使用
tag（字符串）：保存图的名称
scalar_value（浮点型或字符串）：y轴数据（步数）
global_step（int）：x轴数据

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")

x = range(100)
for i in x:
    writer.add_scalar('y=2x', i * 2, i)
writer