设置可训练参数¶
- torchtune.modules.peft.set_trainable_params(model: Module, adapter_params: Union[Dict[str, Any], Set]) None[source]¶
根据适配器参数的状态字典为nn.Module设置可训练参数。
- Parameters:
model (nn.Module) – 包含一些适配器参数的模型类的实例。
adapter_params (Union[Dict[str, Any], Set]) – 状态字典，将适配器键名映射到它们各自的nn.Parameters（即
get_adapter_params()的输出）。
- Returns:
无
