code: def get_state_dict(self, weight_name):
      if not self.is_empty:
          try:
              models = self.models
              keys = [key.replace("best_", "") for key in models.keys()]
              log.info("Available weights : {}".format(keys))
              try:
                  key_name = "best_{}".format(weight_name)
                  model = models[key_name]
                  log.info(
                      "Model loaded from {}:{}.".format(
                          self._check_path, key_name
                      )
                  )
                  return model
              except:
                  key_name = Checkpoint._LATEST
                  model = models[Checkpoint._LATEST]
                  log.info(
                      "Model loaded from {}:{}".format(
                          self._check_path, key_name
                      )
                  )
                  return model
          except:
              raise Exception(
                  "This weight name isn't within the checkpoint "
              )
  
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 7 for calling the Checkpoint obj's underlying member method code with example usage: obj.get_state_dict(weight_name) and returns: model, model