本人使用的是vscode 很简单
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('./logs')writer.add_scalar('train_loss',loss.val(),iteration) # 名字,数据,迭代次数
训练的过程中会产生一个./logs的文件夹,里面存放的是保存的训练loss数据,
然后在命令行运行
tensorboard --logdir './logs'