打印模型结构的问题
收藏
(1)请问一下,要查看ppcls或ppdet保存的模型结构,应该用什么指令或者操作呢?
(2)像下面这种是可以直接通过paddle.summary()进行打印模型结构,那如何打印上面自行设计的通过PaddleClas或PaddleDetection训练生成的网络结构?该如何加载训练模型
import paddle
model = paddle.vision.models.resnet50(pretrained=True)
paddle.summary(model, (64, 3, 224, 224))
0
收藏
请登录后评论
可视化网络结构可以使用飞桨可视化分析工具VisualDL:https://github.com/PaddlePaddle/VisualDL
其中有一个graph功能,可以直接拖拽上传pdmodel文件即可直接查看,AIStudio也有这个可视化接口,可以直接点击选择对应文件进行可视化,不过近期在修复中,预计9.23日重新上线AIStudio,您可以在本地安装使用(pip install visualdl --upgrade),启动命令为 visualdl --model ./log/ppcls.pdmodel (--model后为模型结构文件所在路径)即可可视化
具体使用可见:https://github.com/PaddlePaddle/VisualDL/blob/develop/docs/components/README_CN.md#Graph--%E7%BD%91%E7%BB%9C%E7%BB%93%E6%9E%84%E7%BB%84%E4%BB%B6
优秀
你好,你查看成功了吗