class_docstr: 封装nn.Module, 提供 save 和 load 方法
docstr: 加载指定路径的模型
docstr_tok: ["\u52a0\u8f7d\u6307\u5b9a\u8def\u5f84\u7684\u6a21\u578b"]
code: def load(self, path, device):
      """
      加载指定路径的模型
      """
      self.load_state_dict(torch.load(path, map_location=device))
  
file_path: DeepKE/src/deepke/attribution_extraction/standard/models/BasicModule.py
imports: import os
  import time
  import torch
  import torch.nn
label_desc: Use image node_id 2 for calling the BasicModule obj's underlying member method code with example usage: obj.load(path, device) without return types