openfl.protocols.utils.deconstruct_model_proto

openfl.protocols.utils.deconstruct_model_proto(model_proto, compression_pipeline)

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:
  • tensor_dict – A dictionary where the keys are tensor names and the

  • values are the corresponding tensors.

  • round_number – The round number for the model.