code: def check_checkpoint_args(checkpoint_args):
      """Ensure fixed arguments for a model are the same for the input
      arguments and the one retrieved from checkpoint."""
      args = get_args()
  
      def _compare(arg_name, old_arg_name=None, default=None):
          if old_arg_name is not None:
              ckpt_arg_name = old_arg_name
          else:
              ckpt_arg_name = arg_name
          if default is not None:
              checkpoint_value = getattr(
                  checkpoint_args, ckpt_arg_name, default
              )
          else:
              checkpoint_value = getattr(checkpoint_args, ckpt_arg_name)
          args_value = getattr(args, arg_name)
          error_message = (
              "{} value from checkpoint ({}) is not equal to the "
              "input argument value ({}).".format(
                  arg_name, checkpoint_value, args_value
              )
          )
          assert checkpoint_value == args_value, error_message
  
      _compare("num_layers")
      _compare("hidden_size")
      _compare("num_attention_heads")
      _compare("add_position_embedding", default=True)
      if args.vocab_file:
          _compare("max_position_embeddings")
          _compare("make_vocab_size_divisible_by")
          _compare("padded_vocab_size")
          _compare("tokenizer_type")
      if args.data_parallel_random_init:
          _compare("data_parallel_random_init")
      if get_checkpoint_version() < 3.0:
          _compare(
              "tensor_model_parallel_size",
              old_arg_name="model_parallel_size",
          )
      if get_checkpoint_version() >= 3.0:
          _compare("tensor_model_parallel_size")
          _compare("pipeline_model_parallel_size")
  
imports: import os
  import random
  import sys
  import numpy
  import torch
  from megatron import update_num_microbatches
  from megatron.core import mpu, tensor_parallel
  from .global_vars import get_args
  from .utils import unwrap_model, print_rank_0
label_desc: Use image node_id 3 for calling a global function with example usage: check_checkpoint_args(checkpoint_args) without return types