from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
作用:
1.显示不同步的图像
writer.add_image(''title'', image_array, step)
2.显示数,loss
writer.add_scalar(''title'', y, x)
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
作用:
1.显示不同步的图像
writer.add_image(''title'', image_array, step)
2.显示数,loss
writer.add_scalar(''title'', y, x)