def initialize_computation():
"""Orchestration logic for server model initialization."""
initial_global_model, initial_global_optimizer_state = tff.federated_eval(
server_init, tff.SERVER)
return tff.federated_zip(
ServerState(model=initial_global_model,
optimizer_state=initial_global_optimizer_state,
delta_aggregate_state=aggregation_process.initialize(),
model_broadcast_state=broadcast_process.initialize()))
def initialize_comp():
if not isinstance(stateful_fn.initialize, tff.Computation):
initialize = tff.tf_computation(stateful_fn.initialize)
else:
initialize = stateful_fn.initialize
return tff.federated_eval(initialize, tff.SERVER)
def init():
return tff.federated_eval(create_dataset, tff.CLIENTS)
def init():
return tff.federated_eval(make_zero, tff.SERVER)
def initialize_computation():
"""Orchestration logic for server model initialization."""
return tff.federated_eval(server_init, tff.SERVER)