首页 Paddle框架 帖子详情
在paddle里怎么做到只更新部分参数,另外参数
收藏
快速回复
Paddle框架 问答深度学习模型训练炼丹技巧 2801 3
在paddle里怎么做到只更新部分参数,另外参数
收藏
快速回复
Paddle框架 问答深度学习模型训练炼丹技巧 2801 3

迁移学习,如何微调呢,在paddle里怎么做到只更新部分参数,另外参数不更新

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

https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/Variable_cn.html#stop_gradient

可以看看这个文档,具体的操作我没试过(手动捂脸|)

0
回复
thinc
#3 回复于2020-05

class paddle.fluid.ParamAttr(trainable=True, ...)

冻结参数的方法有很多种,如果指定ParamAttr的trainable参数为False,则会停止该层更新,建议使用这种,其他的一些方法可能导致连续的参数被冻

1
回复
thinc
#4 回复于2020-05
thinc #3
class paddle.fluid.ParamAttr(trainable=True, ...) 冻结参数的方法有很多种,如果指定ParamAttr的trainable参数为False,则会停止该层更新,建议使用这种,其他的一些方法可能导致连续的参数被冻
展开

w1 = paddle.fluid.ParamAttr(trainable=True, ...)

w2 = paddle.fluid.ParamAttr(trainable=False, ...)    # 停止该层更新

b1 = paddle.fluid.ParamAttr(trainable=True, ...)

大概就是这么回事

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