Skip to content

GitLab

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

Expose learning rate schedule classes. 

PiperOrigin-RevId: 369375299
parent cf86fd97
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/modeling/optimization/__init__.py
View file @ 3c46cfb5
......@@ -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.optimizer_config import *
from official.modeling.optimization.ema_optimizer import ExponentialMovingAverage
from official.modeling.optimization.lr_schedule import *
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