可视化loss和acc

参考https://www.jianshu.com/p/46eb3004beca

  1. 环境安装：

    conda activate xxx

    pip install tensorboardX

    pip install tensorflow

  2. 代码：

    from tensorboardXimport SummaryWriter
    writer = SummaryWriter('runs/001')
        writer.add_scalar('Train/Loss', train_loss / batch_idx, epoch)
        writer.add_scalar('Train/Acc', 100.0 * correct / total, epoch)
    writer.close()

  3. 服务器：

    conda activate xxx

    tensorboard --logdir=runs/001

  4. 本地：

    终端上输入：ssh -p 22222 -L 6006:localhost:6006 yinwenbin@192.168.2.237

    浏览器上输入：localhost:6006

可视化模型

参考：https://blog.csdn.net/sunqiande88/article/details/80155925?utm_source=copy

import torchvision.models as models
from tensorboardX import SummaryWriter
import torch
model = models.resnet18()
dummy_input = torch.rand(13, 3, 224, 224)
with SummaryWriter(comment='resnet18') as w:
　　w.add_graph(model, (dummy_input, ))

conda activate xxx

tensorboard --logdir runs

浏览器上输入：localhost:6006

 

若提示错误：'torch._C.Value' object has no attribute 'debugName'

修改tensorboardX 1.9为tensorboardX 1.8

参考：https://blog.csdn.net/East_Plain/article/details/103073311