tensorboard

代码正文中需要write

writer = SummaryWriter("dataloader") # 开始写入内容，这一句需要

# 中间这一堆是要写入的内容，可以替换
step = 0
for data in test_loader:
    imgs,targets = data
    # print(imgs.shape)
    # print(targets)
    writer.add_images("test_data",imgs,step)
    step = step + 1

writer.close()  # 最后关掉，这一句也需要

运行代码之后，在terminal中输入

 tensorboard --logdir="dataloader"

logdir路径的名字，与SummaryWriter一致，都是"dataloader"，否则会找不到。

技巧：
写入的时候，先不用import什么东西，直接SummaryWriter，然后代码中会出现红色的下划线。根据提示选择import，就会自动生成from … import…导入语句了