blazefl.utils.deserialize_model

blazefl.utils.deserialize_model#

blazefl.utils.deserialize_model(model: Module, serialized_parameters: Tensor) None[source]#

Deserialize a flat tensor back into a PyTorch model’s parameters.

Parameters:
  • model (torch.nn.Module) – The PyTorch model to update.

  • serialized_parameters (torch.Tensor) – The tensor containing the parameters.

Returns:

None