code: def load_checkpoint(prefix, epoch):
      """
      Load model checkpoint from file.
      :param prefix: Prefix of model name.
      :param epoch: Epoch number of model we would like to load.
      :return: (arg_params, aux_params)
      arg_params : dict of str to NDArray
          Model parameter, dict of name to NDArray of net's weights.
      aux_params : dict of str to NDArray
          Model parameter, dict of name to NDArray of net's auxiliary states.
      """
      save_dict = mx.nd.load("%s-%04d.params" % (prefix, epoch))
      arg_params = {}
      aux_params = {}
      for k, v in save_dict.items():
          tp, name = k.split(":", 1)
          if tp == "arg":
              arg_params[name] = v
          if tp == "aux":
              aux_params[name] = v
      return arg_params, aux_params
  
label_desc: Use image node_id 1 for calling a global function with example usage: load_checkpoint(prefix, epoch) and returns: arg_params,  aux_params