在pytorch中使用tensorboard

目录

  • 导包
  • 实例化
  • 将模型写入tensorboard
  • 记录数据(在每个epoch的验证阶段)
  • 关闭服务(结束记录)
  • 使用tensorboard查看

导包

from torch.utils.tensorboard import SummaryWriter

实例化

log_dir是写入地址

tb_write=SummaryWriter(log_dir='runs/test')

将模型写入tensorboard

给模型和输入

init_img = torch.zeros((1, 3, 224, 224), device=device)
tb_writer.add_graph(model, init_img)

记录数据(在每个epoch的验证阶段)

三个参数分别是标签名,值,步数

tb_writer.add_scalar(tags[0], mean_loss, epoch)
tb_writer.add_scalar(tags[1], acc, epoch)
tb_writer.add_scalar(tags[2], optimizer.param_groups[0]["lr"], epoch)

关闭服务(结束记录)

tb_write.close()

使用tensorboard查看

比如文件在/Users/liuzao/WorkPlace/wpforPycharm/runs/test里
在test文件下用终端打开,输入如下指令

tensorboard --logdir=./

你可能感兴趣的:(pytorch,人工智能,python)