Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

Module: tff.learning.optimizers

Libraries for optimization algorithms.

Classes

class Optimizer: Represents an optimizer for use in TensorFlow Federated.

Functions

build_adagrad(...): Returns a tff.learning.optimizers.Optimizer for Adagrad.

build_adam(...): Returns a tff.learning.optimizers.Optimizer for Adam.

build_rmsprop(...): Returns a tff.learning.optimizers.Optimizer for RMSprop.

build_sgdm(...): Returns a tff.learning.optimizers.Optimizer for momentum SGD.

build_yogi(...): Returns a tff.learning.optimizers.Optimizer for Yogi.

check_weights_gradients_match(...): Checks that weights and gradients match.

handle_indexed_slices_gradients(...): Converts any tf.IndexedSlices to tensors.