Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tff.learning.framework.ClientOutput

View source on GitHub

Structure for outputs returned from clients during federated optimization.

tff.learning.framework.ClientOutput(
    weights_delta, weights_delta_weight, model_output, optimizer_output
)

Fields:

  • weights_delta: a dictionary of updates to the model's trainable variables.
  • weights_delta_weight: weight to use in a weighted mean when aggregating weights_delta.
  • model_output: a structure matching tff.learning.Model.report_local_outputs, reflecting the results of training on the input dataset.
  • optimizer_output: additional metrics or other outputs defined by the optimizer.