tensboard可视化的问题

Question

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

Answer 1

• 建议你看下这篇博客👉 ：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()
  

  效果：