Internal change

PiperOrigin-RevId: 316052873

official/modeling/optimization/configs/learning_rate_config.py

@@ -146,17 +146,17 @@ class LinearWarmupConfig(base_config.Config):
 
 
 @dataclasses.dataclass
-class PolinomialWarmupConfig(base_config.Config):
+class PolynomialWarmupConfig(base_config.Config):
   """Configuration for linear warmup schedule config.
 
-  This class is a container for the polinomial warmup schedule configs.
+  This class is a container for the polynomial warmup schedule configs.
 
   Attributes:
-    name: The name of warmup schedule. Defaults to Polinomial.
-    power: Polinomial power. Defaults to 1.
+    name: The name of warmup schedule. Defaults to Polynomial.
+    power: Polynomial power. Defaults to 1.
     warmup_steps: Warmup steps. Defaults to None.
   """
-  name: str = 'polinomial'
+  name: str = 'polynomial'
   power: float = 1
   warmup_steps: Optional[int] = None
 

official/modeling/optimization/configs/optimization_config.py

@@ -76,7 +76,7 @@ class WarmupConfig(oneof.OneOfConfig):
   """
   type: Optional[str] = None
   linear: lr_cfg.LinearWarmupConfig = lr_cfg.LinearWarmupConfig()
-  polynomial: lr_cfg.PolinomialWarmupConfig = lr_cfg.PolinomialWarmupConfig()
+  polynomial: lr_cfg.PolynomialWarmupConfig = lr_cfg.PolynomialWarmupConfig()
 
 
 @dataclasses.dataclass

official/modeling/optimization/optimizer_factory_test.py

@@ -259,7 +259,7 @@ class OptimizerFactoryTest(tf.test.TestCase):
     for step, value in expected_lr_step_values:
       self.assertAlmostEqual(lr(step).numpy(), value)
 
-  def test_stepwise_lr_with_polinomial_warmup_schedule(self):
+  def test_stepwise_lr_with_polynomial_warmup_schedule(self):
     params = {
         'optimizer': {
             'type': 'sgd',