首页 Paddle框架 帖子详情
Paddle有相应的API提取卷积核里面的参数吗
收藏
快速回复
Paddle框架 问答深度学习 1021 3
Paddle有相应的API提取卷积核里面的参数吗
收藏
快速回复
Paddle框架 问答深度学习 1021 3

如题所述,在加载模型运行时,有没有办法获取某个隐藏层里卷积核的参数,以及其输出的特征图。(找了下paddle文档没找着)

0
收藏
回复
全部评论(3)
时间顺序
水水水的老师
#2 回复于2020-05

没找到

0
回复
水水水的老师
#3 回复于2020-05

找了一圈  也没在文档中找到

0
回复
m
mar836924042
#4 回复于2020-05
找了一圈  也没在文档中找到

我找了pytorch的资料,然后又尝试了下Paddle,发现也有一样的API。

#假设myCNN()是我用动态图搭建好的一个神经网络类,"model_best"是我的模型参数
with fluid.dygraph.guard():
    model_dict, _ = fluid.load_dygraph('model_best')
    model = MyCNN()
    model.load_dict(model_dict) #加载模型参数
    for param in model.named_parameters():
        print(param[0])
        print(param[1])
        print("----")

param[0]会按顺序逐个输出卷积层1、bias1、BN1层、卷积层2、bias2……的名字,param[1]会输出对应的参数,只不过输出的是Variable的形式,然后再用官网的API提取Variable的值就得到了。

 

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户