TensorBoard(一）

SummaryWriter

引入库

from torch.utils.tensorboard import SummaryWriter

比较常用的两个函数

writer.add_image()
writer.add_scalar()

小tips

• 按ctrl 把鼠标移到SummaryWriter 就可看到这个类的相关介绍
• #按 ctrl+/ 可以直接将其注释掉

首先介绍

1. writer.add_scalar()

    def add_scalar(self, tag, scalar_value, global_step=None, walltime=None):
        """Add scalar data to summary.

        Args:
            tag (string): Data identifier
            scalar_value (float or string/blobname): Value to save
            global_step (int): Global step value to record
            walltime (float): Optional override default walltime (time.time())
              with seconds after epoch of event

tag:题目

#按ctrl 把鼠标移到SummaryWriter 就可看到这个类的相关介绍
from torch.utils.tensorboard import SummaryWriter

writer=SummaryWriter("logs")
#按 ctrl+/ 可以直接将其注释掉
# writer.add_image()
#y=x
for i in range(100):
    writer.add_scalar("y=x",i,i)


writer.close()

可以打开tensorboard的logs

ctrl+c 加上可以直接变化

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

可直接在浏览器中刷新

每次的tag需要变 或者删掉之前的log不然会进行一个累和