code: def on_batch_end(self, batch, logs=None):
      self._restore_momentum_if_needed()
  
imports: from packaging import version
  import warnings
  import horovod.tensorflow
  import tensorflow
label_desc: Use image node_id 8 for calling the LearningRateScheduleCallbackImpl obj's underlying member method code with example usage: obj.on_batch_end(batch, logs) without return types