绘制loss曲线

通过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
        #...

绘图：在终端中这么做

点击网址就可以看到生成的曲线