def get_ckpt_all_steps(checkpoints_dir: str) -> List[int]:
  """Returns a list of available step numbers in ascending order."""
  # Assumes the checkpoint has the format "checkpoint_\d+"
  glob_pattern = os.path.join(checkpoints_dir, 'checkpoint_*')
  checkpoint_paths = gfile.glob(glob_pattern)
  steps = []
  for each in checkpoint_paths:
    steps.append(int(each.split('_')[-1]))
  sorted(steps)
  return steps
Returns a list of available step numbers in ascending order.
videogvt/train_lib/eval_utils.py