code: def _load_model(args, rank=0):
      if "load_epoch" not in args or args.load_epoch is None:
          return (None, None, None)
      assert args.model_prefix is not None
      model_prefix = args.model_prefix
      if rank > 0 and os.path.exists(
          "%s-%d-symbol.json" % (model_prefix, rank)
      ):
          model_prefix += "-%d" % (rank)
      sym, arg_params, aux_params = mx.model.load_checkpoint(
          model_prefix, args.load_epoch
      )
      logging.info(
          "Loaded model %s_%04d.params", model_prefix, args.load_epoch
      )
      return (sym, arg_params, aux_params)
  
file_path: DALI/docs/examples/frameworks/mxnet/demo/common/fit.py
imports: import logging
  import os
  import time
  import re
  import math
  import mxnet
label_desc: Use image node_id 2 for calling a global function with example usage: _load_model(args, rank) and returns: sym,  arg_params,  aux_params, None,  None,  None