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(
":optimizer",
":rmsprop",
":sgdm",
":yogi",
],
)
......
......@@ -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.rmsprop import build_rmsprop
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