提交 dda5f5d5 编辑于 作者: Jakub Konecny's avatar Jakub Konecny 提交者: tensorflow-copybara
浏览文件

Adds Yogi to the API.

PiperOrigin-RevId: 412219459
上级 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
Supports Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册