code: def save(self, epoch=0, cfg=None):
      time_prefix = time.strftime("%Y-%m-%d_%H-%M-%S")
      prefix = os.path.join(cfg.cwd, "checkpoints", time_prefix)
      os.makedirs(prefix, exist_ok=True)
      name = os.path.join(
          prefix, cfg.model_name + "_" + f"epoch{epoch}" + ".pth"
      )
  
      torch.save(self.state_dict(), name)
      return name
  
file_path: DeepKE/example/triple/cnschema/LMModel.py
imports: import os
  import time
  import torch
  import torch.nn
  from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
  from torch import nn
  from transformers import BertModel
  from utils import seq_len_to_mask
label_desc: Use image node_id 3 for calling the LM obj's underlying member method code with example usage: obj.save(epoch, cfg) and returns: name