在paddle里怎么做到只更新部分参数,另外参数
收藏
迁移学习,如何微调呢,在paddle里怎么做到只更新部分参数,另外参数不更新
0
收藏
请登录后评论
https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/Variable_cn.html#stop_gradient
可以看看这个文档,具体的操作我没试过(手动捂脸|)
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, ...)
大概就是这么回事