通过tensorboad绘制:
首先需要通过summary保存绘图需要的loss数据
from torch.utils.tensorboard import SummaryWriter
class Train:
def __init__(self,root, weight_path):
#...
self.summaryWriter=SummaryWriter('logs') #保存在文件夹logs下
#...
def __call__(self):
#...
self.summaryWriter.add_scalar('train_loss',train_loss,step1) #添加变量train_loss进入summary中,步长为step1
#...
绘图:在终端中这么做
点击网址就可以看到生成的曲线