tensorboard的基本使用

1. 加载函数

from torch.utils.tensorboard import SummaryWriter

2. 新建文件夹

os.makedirs(f'{output_dir}/log', exist_ok=True)

3. 初始化

writer = SummaryWriter('./log')

4. 循环画曲线

writer.add_scalar('loss', loss, iter)

5. 打开指令

tensorboard --logdir=./log --host=127.0.0.1