code: def _build_global_state(self):
      sd = torch.load(
          self.mp_rank_files[0], map_location=torch.device("cpu")
      )
      self.global_state[ITERATION_KEY] = sd.get(ITERATION_KEY, 0)
      self.global_state[ARGS_KEY] = sd.get(ARGS_KEY, None)
  
imports: import os
  from typing import Dict
  import torch
  from .reshape_3d_utils import model_3d_desc
  from .reshape_utils import basic_folder_validation, merge_state, partition_data, get_files, get_files_with_prefix
  from .constants import MODEL_FILE_PREFIX, LAYER_FILE_PREFIX
  from .reshape_meg_2d import reshape_meg_2d_parallel, meg_2d_parallel_map
  from .zero_checkpoint import ZeROCheckpoint
  from .constants import *
label_desc: Use image node_id 10 for calling the DeepSpeedCheckpoint obj's underlying member method code with example usage: obj._build_global_state() without return types