【pytorch】模型可视化

针对pytorch有很多可视化方式，比如，tensorboard_logger、visdom等，但这些比较复杂和不够友好，则使用更强大的工具--tenorboardX

from torch.utils.tensorboard import SummaryWriter
if __name__ == "__main__":
    input = torch.Tensor(2, 3, 112, 112)
    net = SEResNet34_IR()
    with SummaryWriter(log_dir='logs', comment='Net') as W:
       W.add_graph(net, (input, ))

 

如何使用？

# cd 到logs上级目录上，在win10上cmd的pytorch环境下输入
tensorboard --logdir=logs --port 6066

然后打开：http://localhost:6066/

 

更新【2020年8月27】

怎么都显示不出来ÿ