code: def get_lr_schedule_callbacks(initial_lr):
      import horovod.tensorflow.keras as hvd
  
      return [
          hvd.callbacks.LearningRateWarmupCallback(
              warmup_epochs=5, initial_lr=initial_lr, verbose=True
          ),
          # Horovod: after the warmup reduce learning rate by 10 on the 30th, 60th and 80th epochs.
          hvd.callbacks.LearningRateScheduleCallback(
              start_epoch=5,
              end_epoch=30,
              multiplier=1.0,
              initial_lr=initial_lr,
          ),
          hvd.callbacks.LearningRateScheduleCallback(
              start_epoch=30,
              end_epoch=60,
              multiplier=1e-1,
              initial_lr=initial_lr,
          ),
          hvd.callbacks.LearningRateScheduleCallback(
              start_epoch=60,
              end_epoch=80,
              multiplier=1e-2,
              initial_lr=initial_lr,
          ),
          hvd.callbacks.LearningRateScheduleCallback(
              start_epoch=80, multiplier=1e-3, initial_lr=initial_lr
          ),
          LRLogger(),
      ]
  
imports: from zoo.orca import init_orca_context, stop_orca_context
  import os
  import argparse
  import tensorflow
label_desc: Use image node_id 18 for calling a global function with example usage: get_lr_schedule_callbacks(initial_lr) and returns: list