huggingface被禁了，改成从本地加载bert，一直报错FileNotFoundError: [Errno 2] No such file or directory: 'params.pkl'

[nlper01]

如何解决？我修改了casrel.py中的代码

class CasRel(nn.Module): def __init__(self, config): super(CasRel, self).__init__() self.config = config self.bert_dim = 768 model_path = r'C:\Users\visitor_zxd\Desktop\新建文件夹\CasRel-pytorch-reimplement\CasRel-reimplement\bert-base-chinese' self.bert_encoder = BertModel.from_pretrained(model_path) # model_name = 'bert-base-chinese' # # model_name = "bert-base-multilingual-cased" # self.bert_encoder = BertModel.from_pretrained(model_name) self.sub_start_tageer = nn.Linear(self.bert_dim, 1) self.sub_end_tagger = nn.Linear(self.bert_dim, 1) self.obj_start_tagger = nn.Linear(self.bert_dim, config['relation_types']) self.obj_end_tagger = nn.Linear(self.bert_dim, config['relation_types'])

[Sherlock-coder]

注释掉train.py中的model.load_state_dict(torch.load('params.pkl'))即可