利用Tensorboard可视化

  1. from torch.utils.tensorboard import SummaryWriter
  2. writer = SummaryWriter('./data/tensorboard')
  3. writer.add_graph(net,input_to_model = torch.rand(10,2))
  4. writer.close()

将网络结构通过SummaryWriter 写到本地路径当中
命令行里使用

tensorboard --logdir logpath进行网络结构可视化

notebook里面使用：

  1. from tensorboard import notebook
  3. notebook.list()
  5. notebook.start('--logdir p')