如何冻结模型的部分层

[sherrijj]

如果要冻结预训练好的backbone或其中部分层，应该怎么设置？目前的代码好像是整个模型参数都要被训练。

[muzi0111]

backbone 在模型里统一叫 emb_func，优化器初始化时会读取 config["optimizer"]["other"]，可以在对应部分将参数设置为想要的学习率，设置为none或者～则冻结参数（参考trainer.py:_init_optim)，举例：
optimizer:
name: Adam
kwargs:
lr: 1e-3
other:
emb_func: ~ #冻结了backbone的参数

[sherrijj]

非常感谢！ 在 2026-03-09 20:16:01，"muzi0111" ***@***.***> 写道： muzi0111 left a comment (RL-VIG/LibFewShot#131) backbone 在模型里统一叫 emb_func，优化器初始化时会读取 config["optimizer"]["other"]，可以在对应部分将参数设置为想要的学习率，设置为none或者～则冻结参数（参考trainer.py:_init_optim)，举例： optimizer: name: Adam kwargs: lr: 1e-3 other: emb_func: ~ #冻结了backbone的参数 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>