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.