nvflare.app_opt.pt.utils module

feed_vars(model: Module, model_params)

Feed variable values from model_params to pytorch state_dict.

Parameters:

• model (nn.Module) – the local pytorch model

• model_params – a ModelData message

Returns:

a list of params and a dictionary of vars to params