本人使用的是vscode 很簡單
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('./logs')writer.add_scalar('train_loss',loss.val(),iteration) # 名字,數據,迭代次數
訓練的過程中會產生一個./logs的文件夾,里面存放的是保存的訓練loss數據,
然后在命令行運行
tensorboard --logdir './logs'