get_fusion_params

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

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

参数:

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

返回:

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

适配器参数。

返回类型:

Dict[str, nn.Parameter]