使用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'