记录跟着小土堆学习pytorch的代码及注释。
代码:
`SummaryWriter` 类提供了一个高级 API,用于在给定目录中创建事件文件,并向其中添加摘要和事件。 该类异步更新文件内容。
from torch.utils.tensorboard import SummaryWriter
#创建类的实例
#将对应的事件文件存储到logs文件夹
writer = SummaryWriter("logs")
#使用到两个方法,add_scalar是标量的意思
# writer.add_image()
# y = 2x
for i in range(100):
# ("标题",y轴,x轴)
writer.add_scalar("y=2x", 2*i, i)
writer.close()