code: def save_checkpoint(
      prefix,
      epoch,
      symbol,
      arg_params,
      aux_params,
      remove_amp_cast=True,
  ):
      """Checkpoint the model data into file.
  
      Parameters
      ----------
      prefix : str
          Prefix of model name.
      epoch : int
          The epoch number of the model.
      symbol : Symbol
          The input Symbol.
      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.
      remove_amp_cast : bool, optional
          Whether to remove the amp_cast and amp_multicast operators, before saving the model.
      Notes
      -----
      - ``prefix-symbol.json`` will be saved for symbol.
      - ``prefix-epoch.params`` will be saved for parameters.
      """
      if symbol is not None:
          symbol.save(
              f"{prefix}-symbol.json", remove_amp_cast=remove_amp_cast
          )
  
      save_dict = {
          (f"arg:{k}"): v.as_in_context(cpu())
          for k, v in arg_params.items()
      }
      save_dict.update(
          {
              (f"aux:{k}"): v.as_in_context(cpu())
              for k, v in aux_params.items()
          }
      )
      param_name = f"{prefix}-{epoch:04}.params"
      nd.save(param_name, save_dict)
      logging.info('Saved checkpoint to "{}"'.format(param_name))
  
file_path: mxnet/python/mxnet/model.py
imports: import os
  import logging
  from collections import namedtuple
  import numpy
  from .None import ndarray
  from .None import symbol
  from .None import kvstore
  from .device import cpu
label_desc: Use image node_id 7 for calling a global function with example usage: save_checkpoint(prefix, epoch, symbol, arg_params, aux_params, remove_amp_cast) without return types