pytorch的可视化

利用Tensorboard可视化

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

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

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

notebook里面使用：

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