get_fusion_params

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

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

参数:

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

返回值:

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

仅包含 Adapter 参数。

返回类型:

Dict[str, nn.Parameter]