【Tensorboard 使用】生成model结构图解决方案:add_graph() 和 torchsummary可视化模型信息
文章目录一、add_graph()1.具体参数及介绍见之前博客——可视化工具Tensorboard2.pytorch源码3.可视化操作步骤:Tensorboard的打开方式见之前可视化损失函数的步骤二、torchsummary1.具体参数及介绍见之前博客——可视化工具Tensorboard2.pytorch源码3.打印结果参考深度之眼pytorch框架班一、add_graph()1.具体参数及介绍