class_docstr: official.nlp.optimization.WarmUp
code: def get_config(self):
      return {  # pragma: no cover
          "initial_learning_rate": self.initial_learning_rate,
          "decay_schedule_fn": self.decay_schedule_fn,
          "warmup_steps": self.warmup_steps,
          "power": self.power,
          "name": self.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 3 for calling the WarmUp obj's underlying member method code with example usage: obj.get_config() and returns: dict