code: def get_current_long_cycle_shape(schedule, epoch):
      """
      Given a schedule and epoch index, return the long cycle base shape.
      Args:
          schedule (configs): configs that contains training and multigrid specific
              hyperparameters. Details can be seen in
              slowfast/config/defaults.py.
          cur_epoch (int): current epoch index.
      Returns:
          shapes (list): A list describing the base shape in a long cycle:
              [batch size relative to default,
              number of frames, spatial dimension].
      """
      for s in schedule:
          if epoch < s[-1]:
              return s[1]
      return schedule[-1][1]
  
imports: import numpy
  import slowfast.utils.logging
label_desc: Use image node_id 2 for calling a global function with example usage: get_current_long_cycle_shape(schedule, epoch) and returns: schedule, s