code: def load_model(args, n_entities, n_relations, ckpt=None):
      model = KEModel(
          args,
          args.model_name,
          n_entities,
          n_relations,
          args.hidden_dim,
          args.gamma,
          double_entity_emb=args.double_ent,
          double_relation_emb=args.double_rel,
      )
      if ckpt is not None:
          assert (
              False
          ), "We do not support loading model emb for genernal Embedding"
  
      print("Load model {}".format(args.model_name))
      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 1 for calling a global function with example usage: load_model(args, n_entities, n_relations, ckpt) and returns: model