class_docstr: official.nlp.optimization.WarmUp
code: def __init__(
      self,
      initial_learning_rate,
      decay_schedule_fn,
      warmup_steps,
      power=1.0,
      name=None,
  ):
      super(WarmUp, self).__init__()
      self.initial_learning_rate = initial_learning_rate
      self.warmup_steps = warmup_steps
      self.power = power
      self.decay_schedule_fn = decay_schedule_fn
      self.name = name
  
imports: from typing import List
  import tensorflow
  from tensorflow import keras
  from tensorflow import nest
  from tensorflow.keras import layers
  from tensorflow.keras.layers.experimental import preprocessing
  from autokeras.utils import data_utils
label_desc: Use image node_id 1 to create a new WarmUp object from inherited base classes: keras.optimizers.schedules with example: obj = WarmUp(initial_learning_rate, decay_schedule_fn, warmup_steps, power, name)