GitHub

tff.learning.state_with_new_model_weights

Returns a ServerState with updated model weights.

tff.learning.state_with_new_model_weights(
    server_state: tff.learning.framework.ServerState,
    trainable_weights: List[np.ndarray],
    non_trainable_weights: List[np.ndarray]
) -> tff.learning.framework.ServerState

Used in the notebooks

Used in the tutorials
Federated Learning for Text Generation

Args
`server_state`	A server state object returned by an iterative training process like `tff.learning.build_federated_averaging_process`.
`trainable_weights`	A list of `numpy` values in the order of the original model's `trainable_variables`.
`non_trainable_weights`	A list of `numpy` values in the order of the original model's `non_trainable_variables`.

Returns
A new server `ServerState` object which can be passed to the `next` method of the iterative process.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2022-03-11 UTC.

tff.learning.state_with_new_model_weights

Used in the notebooks

Args

Returns