docstr:     The API to load the model.
  
  Parameters
  ----------
  load_path : str
      The root path to load the model.
  checkpoint_name : str
      The name of the model to be loaded.
  
  Returns
  -------
  Graph2XBase
docstr_tok: ["The","API","to","load","the","model",".","Parameters","--","--","--","--","--","load_path",":","str","The","root","path","to","load","the","model",".","checkpoint_name",":","str","The","name","of","the","model","to","be","loaded",".","Returns","--","--","--","-","Graph2XBase"]
code: def load_checkpoint(cls, load_path, checkpoint_name):
      """
          The API to load the model.
  
      Parameters
      ----------
      load_path : str
          The root path to load the model.
      checkpoint_name : str
          The name of the model to be loaded.
  
      Returns
      -------
      Graph2XBase
      """
      checkpoint_path = os.path.join(load_path, checkpoint_name)
      model = torch.load(checkpoint_path)
      return model
  
imports: import os
  import torch
  import torch.backends.cudnn
  import torch.multiprocessing
  import torch.nn
  import torch.nn.functional
  from graph4nlp.pytorch.modules.graph_construction import NodeEmbeddingBasedRefinedGraphConstruction
  from graph4nlp.pytorch.modules.graph_construction.node_embedding_based_graph_construction import NodeEmbeddingBasedGraphConstruction
  from graph4nlp.pytorch.modules.graph_embedding_initialization.embedding_construction import WordEmbedding
  from graph4nlp.pytorch.modules.graph_embedding_initialization.graph_embedding_initialization import GraphEmbeddingInitialization
  from graph4nlp.pytorch.modules.graph_embedding_learning.gat import GAT
  from graph4nlp.pytorch.modules.graph_embedding_learning.gcn import GCN
  from graph4nlp.pytorch.modules.graph_embedding_learning.ggnn import GGNN
  from graph4nlp.pytorch.modules.graph_embedding_learning.graphsage import GraphSAGE
  from graph4nlp.pytorch.modules.prediction.classification.node_classification import BiLSTMFeedForwardNN
label_desc: Use image node_id 6 for calling the Word2tag obj's underlying member method code with example usage: obj.load_checkpoint(cls, load_path, checkpoint_name) and returns: model