fl4health.utils.peft_parameter_extraction module

get_all_peft_parameters_from_model(model)

Function to extract peft parameters associated with a pytorch module. These values are converted from numpy arrays into a Flower Parameters object.

Parameters:

model (nn.Module) – PyTorch model whose parameters are to be extracted

Returns:

Flower Parameters object containing all of the target models state.

Return type:

Parameters