pytorch中tensorboard数据显示

最新推荐文章于 2024-07-05 16:01:35 发布

RNG_uzi_

最新推荐文章于 2024-07-05 16:01:35 发布

阅读量1k

点赞数

文章标签：深度学习 tensorflow

# 解释权归本文作者所有

本文链接：https://blog.csdn.net/RNG_uzi_/article/details/106813316

版权

3D点云专栏收录该内容

28 篇文章 30 订阅

订阅专栏

pytorch 利用tensorboard显示loss，acc曲线等

运行环境：

python3.6.9
pytorch1.13.1
cuda10.0
cudnn7.5.1

tensorboard显示

运行PointRCNN算法进行training，得出events.out.tfevents.1592297776.hkd-Precision-7920-Tower

打开终端输入：tensorboard --logdir path/to/tensorboard_logs/
会有输出：TensorBoard 1.6.0 at http://iccd:6006 (Press CTRL+C to quit)
将上述链接复制到浏览器中打开便可以显示该训练参数（tensorboard）

在这里插入图片描述

tensorboard记录

from tensorboard_logger import Logger

logger = Logger(logdir="./tensorboard_logs", flush_secs=10)
...
def train(net, optimizer):
    for epoch in range(epoch_nums):
        net.train()
        for batch_idx, (inputs, targets) in enumerate(trainloader):           
            inputs = Variable(inputs, requires_grad=True).cuda()
            targets = targets.cuda()
            optimizer.zero_grad()
            outputs = net(inputs)
            loss = criterion(outputs, targets)
            loss.backward()  
            optimizer.step()
            train_loss += loss.item()
            ...
            # 记录所需的变量
            logger.log_value('avg_loss', train_loss/(batch_idx+1), epoch*len(trainloader) + batch_idx)
            logger.log_value('loss', loss.item(), epoch*len(trainloader) + batch_idx)
            logger.log_value('acc', 100. * correct / total, epoch*len(trainloader) + batch_idx)