案例带你学Pytorch（5）——TensorboardX详解（2）

SummaryWriter详解之add_graph示例分析

上一篇分析了SummaryWriter主要函数，本篇借助TensorboardX官方demo，解释add_graph用法

add_graph参数

• add_graph(self, model, input_to_model=None, verbose=False）
• model： model (torch.nn.Module): model to draw
  对于model = Linear（），参数形式直接用model，不加括号
  否则需要用Linear（），否则需要加括号
• input_to_model：input_to_model (torch.Tensor or list of torch.Tensor): a variable or a tuple of variables to be fed.模型输入，可以有一个或者多个。搞不清楚shape时，建议用debug模式观察一下。模型输入必须存在。
• verbose：字面意思是冗长的、啰嗦的。bool变量，默认为False
• 启动tensorboard时，–logdir指向的文件夹下，将会递归所有所有文件（存储在子文件夹中不影响）
• tensorboard --logdir runs有时不工作，可以使用tensorboard --logdir runs --host localhost
• 采用with SummaryWriter() as writer:写法，可以省掉writer.close()