快捷方式

set_trainable_params

torchtune.modules.peft.set_trainable_params(model: Module, adapter_params: Dict[str, Any]) None[源代码]

根据适配器参数的状态字典设置 nn.Module 的可训练参数。

参数:
  • model (nn.Module) – 包含一些适配器参数的模型类实例。

  • adapter_params (Dict[str, Any]) – 状态字典,将适配器键名映射到其相应的 nn.Parameters(即 get_adapter_params() 的输出)。

返回值:

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源