tensorboard 使用

阿奘發表於2024-07-18
  1. 載入函式
from torch.utils.tensorboard import SummaryWriter
  1. 新建資料夾
os.makedirs(f'{output_dir}/log', exist_ok=True)
  1. 初始化
writer = SummaryWriter('./log')
  1. 迴圈畫曲線
writer.add_scalar('loss', loss, iter)
  1. 開啟指令
tensorboard --logdir=./log --host=127.0.0.1

相關文章