Commit dda5f5d5 authored by Jakub Konecny's avatar Jakub Konecny Committed by tensorflow-copybara
Browse files

Adds Yogi to the API.

PiperOrigin-RevId: 412219459
parent 6bfc93a3
...@@ -76,6 +76,7 @@ py_library( ...@@ -76,6 +76,7 @@ py_library(
":optimizer", ":optimizer",
":rmsprop", ":rmsprop",
":sgdm", ":sgdm",
":yogi",
], ],
) )
......
...@@ -20,3 +20,4 @@ from tensorflow_federated.python.learning.optimizers.optimizer import handle_ind ...@@ -20,3 +20,4 @@ from tensorflow_federated.python.learning.optimizers.optimizer import handle_ind
from tensorflow_federated.python.learning.optimizers.optimizer import Optimizer from tensorflow_federated.python.learning.optimizers.optimizer import Optimizer
from tensorflow_federated.python.learning.optimizers.rmsprop import build_rmsprop from tensorflow_federated.python.learning.optimizers.rmsprop import build_rmsprop
from tensorflow_federated.python.learning.optimizers.sgdm import build_sgdm from tensorflow_federated.python.learning.optimizers.sgdm import build_sgdm
from tensorflow_federated.python.learning.optimizers.yogi import build_yogi
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment