nvflare.app_opt.pt.utils module

feed_vars(model: Module, model_params)[source]

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