Skip to content

GitLab

Commit b51b9342 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Expose learning rate schedule classes. 

PiperOrigin-RevId: 369375299
parent e187cef7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/modeling/optimization/__init__.py
View file @ b51b9342
...@@ -19,4 +19,5 @@ from official.modeling.optimization.configs.learning_rate_config import * ...@@ -19,4 +19,5 @@ from official.modeling.optimization.configs.learning_rate_config import *
from official.modeling.optimization.configs.optimization_config import * from official.modeling.optimization.configs.optimization_config import *
from official.modeling.optimization.configs.optimizer_config import * from official.modeling.optimization.configs.optimizer_config import *
from official.modeling.optimization.ema_optimizer import ExponentialMovingAverage from official.modeling.optimization.ema_optimizer import ExponentialMovingAverage
from official.modeling.optimization.lr_schedule import *
from official.modeling.optimization.optimizer_factory import OptimizerFactory from official.modeling.optimization.optimizer_factory import OptimizerFactory
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