docstr: Saves checkpoint with updated mdoels for the given stage
          
docstr_tok: ["Saves","checkpoint","with","updated","mdoels","for","the","given","stage"]
code: def save_objects(
      self,
      models_to_save: Dict[str, Any],
      stage,
      current_stat,
      optimizer,
      schedulers,
      **kwargs,
  ):
      """Saves checkpoint with updated mdoels for the given stage"""
      self.models = models_to_save
      self.optimizer = (
          optimizer.__class__.__name__,
          optimizer.state_dict(),
      )
      self.schedulers = {
          scheduler_name: [
              scheduler.scheduler_opt,
              scheduler.state_dict(),
          ]
          for scheduler_name, scheduler in schedulers.items()
      }
      to_save = kwargs
      for key, value in self.__dict__.items():
          if not key.startswith("_"):
              to_save[key] = value
      torch.save(to_save, self.path)
  
imports: from typing import Any, Dict, List, Optional, Tuple, Union
  import os
  import torch
  import logging
  import copy
  import glob
  import shutil
  from omegaconf import OmegaConf
  from omegaconf import DictConfig
  from torch_points3d.models import model_interface
  from torch_points3d.utils.colors import COLORS, colored_print
  from torch_points3d.core.schedulers.lr_schedulers import instantiate_scheduler
  from torch_points3d.core.schedulers.bn_schedulers import instantiate_bn_scheduler
  from torch_points3d.models.model_factory import instantiate_model
label_desc: Use image node_id 2 for calling the Checkpoint obj's underlying member method code with example usage: obj.save_objects(models_to_save, stage, current_stat, optimizer, schedulers) without return types