Skip to content

GitLab

Commit 132d991f authored by Mark Daoust's avatar Mark Daoust Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 444154402
parent ccacb666
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/modeling/optimization/optimizer_factory.py
View file @ 132d991f
...@@ -86,6 +86,8 @@ class OptimizerFactory: ...@@ -86,6 +86,8 @@ class OptimizerFactory:
(4) Build optimizer. (4) Build optimizer.
This is a typical example for using this class: This is a typical example for using this class:
```
params = { params = {
'optimizer': { 'optimizer': {
'type': 'sgd', 'type': 'sgd',
...@@ -105,6 +107,7 @@ class OptimizerFactory: ...@@ -105,6 +107,7 @@ class OptimizerFactory:
opt_factory = OptimizerFactory(opt_config) opt_factory = OptimizerFactory(opt_config)
lr = opt_factory.build_learning_rate() lr = opt_factory.build_learning_rate()
optimizer = opt_factory.build_optimizer(lr) optimizer = opt_factory.build_optimizer(lr)
```
""" """
def __init__(self, config: opt_cfg.OptimizationConfig): def __init__(self, config: opt_cfg.OptimizationConfig):
......
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