set_trainable_params¶
- torchtune.modules.peft.set_trainable_params(model: Module, adapter_params: Union[Dict[str, Any], Set]) None[源代码]¶
根据适配器参数的状态字典设置 nn.Module 的可训练参数。
- 参数:
model (nn.Module) – 包含某些适配器参数的模型类实例。
adapter_params (Union[Dict[str, Any], Set]) – 将适配器键名映射到其各自 nn.Parameters 的状态字典 (即
get_adapter_params()的输出)。
- 返回:
None
