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
