在jupyter中依次执行下列代码
from torch.utils.tensorboard import SummaryWriter
import numpy as np
writer = SummaryWriter('./log/')
%load_ext tensorboard
for i in range(100):
//add_scalar绘制单个
writer.add_scalar("sinx",np.sin(i),i)
writer.add_scalar("cosx",np.cos(i),i)
//add_scalars将多个绘制到一个图像上
writer.add_scalars(main_tag="curve~x",tag_scalar_dict={
"sinx":np.sin(i),
"cosx":np.cos(i)
},global_step=i)
%tensorboard --logdir=log
整体效果
每个小图的效果
for i in range(100):
//这种格式绘制效果如下图
writer.add_scalar("x/sinx",np.sin(i),i)
writer.add_scalar("x/cosx",np.cos(i),i)
for i in range(100):
writer.add_scalars(main_tag="y/x1",tag_scalar_dict={
"sinx":np.sin(i),
"cosx":np.cos(i)
},global_step=i)
writer.add_scalars(main_tag="y/x2",tag_scalar_dict={
"tanx":np.tan(i),
"cosx":i*np.cos(i)
},global_step=i)