def return_global_model(self):
"""
Serializes the current global torch model and sends it back to the worker.
Returns
----------
dict:
The model dictionary as per the specification in DCFSever
"""
logger.info(f"Example Global Model: returning global model")
model_data = io.BytesIO()
torch.save(self.global_model, model_data)
return create_model_dict(model_data.getvalue(),
self.global_model_version)
def test_ret_global_model_cb():
return create_model_dict(msgpack.packb("Pickle dump of a string"),
global_model_version)