tensboard可视化的问题

tensorboard可视化的问题,怎们激活inactive这里的功能,主要是想用Graph.

img

  • 建议你看下这篇博客👉 :tensorboard绘制graph
  • 除此之外, 这篇博客: 详解Tensorboard及使用教程中的 网络(graph) 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  • add_graph(model, input_to_model=None, verbose=False, use_strict_trace = True)

    参数:

    • model( torch.nn.Module ) – 要绘制的模型。
    • input_to_model ( torch.Tensor or list of torch.Tensor ) – 要输入的变量或变量元组
    • verbose(bool)– 是否在控制台中打印图形结构。
    • use_strict_trace ( bool ) – 是否将关键字参数严格传递给 torch.jit.trace。当您希望跟踪器记录您的可变容器类型(列表、字典)时传递 False.

    例子:

    from torch.utils.tensorboard import SummaryWriter
    import numpy as np
    import torch
    import torch.nn as nn
    
    writer = SummaryWriter()
    class MLP(nn.Module) :
        def __init__(self):
            super(MLP, self).__init__()
            self.Net = nn.Sequential(
                nn.Linear(784, 512),
                nn.ReLU(),
    
                nn.Linear(512, 128),
                nn.ReLU(),
    
                nn.Linear(128, 10)
            )
        def forward(self, input):
            input = input.view(-1, 28 * 28)
            return self.Net(input)
    model = MLP()
    input = torch.FloatTensor(np.random.rand(32, 1, 28, 28))
    writer.add_graph(model, input)
    
    writer.close()
    

    效果:
    在这里插入图片描述
    在这里插入图片描述