tensorboard查看训练中的loss,acc等

使用tensorboard查看全局loss,精度等非常简单,具体如下:

from torch.utils.tensorboard import SummaryWriter
# 设置一个生成log的文件夹
writer = SummaryWriter('/zxy/code/tensorboard_logs/tensorboard_logs_0825')
# 添加想查看的变量
writer.add_scalar('lr', lr_scheduler.get_lr()[0], epoch)
writer.add_scalar('cls_train_loss', cls_loss, epoch)
writer.add_scalar('pose_train_loss', pose_loss, epoch)
writer.add_scalar('total_train_loss', total_loss, epoch)
writer.add_scalar('acc', acc, epoch)

添加以上代码即可,查看时在命令行运行

tensorboard --logdir './zxy/code/tensorboard_logs/tensorboard_logs_0825'

在这里插入图片描述
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

我现在强的可怕~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值