快捷方式

get_fusion_params

torchtune.modules.model_fusion.get_fusion_params(model: Module) Dict[str, Parameter][源代码]

返回模型中对应于融合模块的子集参数。假定任何融合类都已定义 fusion_params() 方法。

参数::

model (nn.Module) – 包含一些融合参数的模型类实例。

返回值::

包含模型状态字典的子集,其中

只有适配器参数。

返回类型::

Dict[str, nn.Parameter]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源