tff.learning.templates.reject_non_finite_update

Rejects the update if any non-finite value is in the update.

tff.learning.templates.reject_non_finite_update(
    state: Any, update: Any
) -> tuple[tf.Tensor, _MeasurementsType]

This is the default should_reject_update function used in build_apply_optimizer_finalizer.

Args
`state`	Unused optimzier state.
`update`	The update to be applied to the model's weights by the optimizer.

Returns
A tuple of: should_reject (bool tensor): True if the update should be rejected, False otherwise. measurements (OrderedDict): A dict with a single key (`update_non_finite`) an an integer tensor of whether the update was rejected.

Returns

A tuple of:

should_reject (bool tensor): True if the update should be rejected, False otherwise.
measurements (OrderedDict): A dict with a single key (update_non_finite) an an integer tensor of whether the update was rejected.

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 2024-09-20 UTC.