tensorboard的基本使用
- 加载函数
from torch.utils.tensorboard import SummaryWriter
- 新建文件夹
os.makedirs(f'{output_dir}/log', exist_ok=True)
- 初始化
writer = SummaryWriter('./log')
- 循环画曲线
writer.add_scalar('loss', loss, iter)
- 打开指令
tensorboard --logdir=./log --host=127.0.0.1