code: def get_learning_rate_fn():
      decay_steps = int(8e6)
      start_decay_step = int(30.0e6)
      warmup_steps = int(150000)
      if FLAGS.debug:
          decay_steps = 30000
          start_decay_step = 20000
          warmup_steps = 0
      return learning_rate_schedule.WarmupDelayedCosineDecay(
          initial_learning_rate=tf.cast(0.0001, dtype=tf.float32),
          constant_learning_rate=tf.cast(
              FLAGS.learning_rate, dtype=tf.float32
          ),
          end_learning_rate=FLAGS.learning_rate / 1000000.0,
          warmup_steps=warmup_steps,
          start_decay_step=start_decay_step,
          decay_steps=decay_steps,
      )
  
imports: import os
  import pickle
  import time
  from absl import app
  from absl import flags
  from absl import logging
  import matplotlib.pyplot
  import numpy
  import tensorflow
  import tensorflow_graphics.projects.points_to_3Dobjects.data_preparation.extract_protos
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import collision_loss
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import cross_entropy_loss
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import focal_loss
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import focal_loss_sparse
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import learning_rate_schedule
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import regression_huber_loss
  from tensorflow_graphics.projects.points_to_3Dobjects.losses import regression_l1_loss
  import tensorflow_graphics.projects.points_to_3Dobjects.models.centernet_vid
  from tensorflow_graphics.projects.points_to_3Dobjects.transforms import transforms_factory
  from tensorflow_graphics.projects.points_to_3Dobjects.utils import evaluator
  import tensorflow_graphics.projects.points_to_3Dobjects.utils.io
  import tensorflow_graphics.projects.points_to_3Dobjects.utils.logger
  import tensorflow_graphics.projects.points_to_3Dobjects.utils.plot
  import tensorflow_graphics.projects.points_to_3Dobjects.utils.tf_utils
  from google3.pyglib import gfile
label_desc: Use image node_id 5 for calling a global function with example usage: get_learning_rate_fn() and returns: learning_rate_schedule