Function - deconstruct_model_proto

Function - deconstruct_model_proto#

openfl.protocols.utils.deconstruct_model_proto(model_proto, compression_pipeline)[source]#

Deconstruct model proto.

This function takes a model protobuf and a compression pipeline, and deconstructs the protobuf into a dictionary of tensors and a round number.

Parameters:
  • model_proto – The protobuf of the model.

  • compression_pipeline – The compression pipeline for the model.

Returns:

A dictionary where the keys are tensor names and the values are the corresponding tensors. round_number: The round number for the model.

Return type:

tensor_dict