code: def load_model_from_checkpoint(
      args, n_entities, n_relations, ckpt_path
  ):
      model = load_model(args, n_entities, n_relations)
      model.load_emb(ckpt_path, args.dataset)
      return model
  
file_path: ogb/examples/lsc/wikikg90m/dgl-ke-ogb-lsc/python/dglke/train_mxnet.py
imports: import mxnet
  from mxnet import gluon
  from mxnet import ndarray
  import os
  import logging
  import time
  import json
  from .models import KEModel
label_desc: Use image node_id 2 for calling a global function with example usage: load_model_from_checkpoint(args, n_entities, n_relations, ckpt_path) and returns: model