code: def __init__(self, warmup_steps=10000, learning_rate_start=1e-1):
      super(CustomSchedule, self).__init__()
      self.warmup_steps = warmup_steps
      self.learning_rate_start = learning_rate_start
  
imports: import os
  import pickle
  from typing import Callable
  from typing import Dict, Union
  import numpy
  import tensorflow
  from headliner.model.bert_model import Transformer, create_masks
  from headliner.model.summarizer import Summarizer
  from headliner.preprocessing.bert_vectorizer import BertVectorizer
  from headliner.preprocessing.preprocessor import Preprocessor
  from headliner.utils.logger import get_logger
label_desc: Use image node_id 1 to create a new CustomSchedule object from inherited base classes: tf.keras.optimizers.schedules with example: obj = CustomSchedule(warmup_steps, learning_rate_start)