很久没有学了,期间打点滴打了一个多星期,太累了,再加上学了一下Python语法基础,再终于开始重新学习pytorch
tensorboard 的使用
import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
for i in range(100):writer.add_scalar("y = 2x", 2*i,i)
# writer.add_scalar("image", torch.randn(1, 3, 224, 224))
writer.close()
在当前文件夹下的终端输入 tensorboard --logdir=E:\pytorch_learn\logs
writer.add_scalar("y = 2x", 2*i,i)
第一个参数是名字,第二个是y轴,第三个是x轴
writer.add_image("test", img_array, 2,dataformats="HWC")这个是显示图像 第一个参数是名字,第二个是图像,第三个是训练的步数,第四个是根据第二个参数的图像格式来决定