1
from torch.utils.tensorboard import SummaryWriter
2
writer = SummaryWriter("logs","comment") # 创建一个日志记录保存的文件夹log,文件后缀名可以通过comment添加。

1
 def add_scalar(self,
2
              tag: str,
3
              scalar_value: Any,
4
              global_step: int = None,
5
              walltime: float = None,
6
              new_style: bool = False,
7
              double_precision: bool = False) -> None
8
# 例子：
9
for i in range(100):
10
   writer.add_scalar("y=x", i, i)