解决PyTorch Lightning报错KeyError: ‘hidden_states’
问题描述:使用解决PyTorch Lightning报错KeyError: ‘hidden_states’。
model = BertModel.from_pretrained('bert-base-uncased')
解决:将以上代码后面增加一个参数,config=BertConfig.from_pretrained(‘bert-base-uncased’,output_hidden_states=True),改为:
model = BertModel.from_pretrained('bert-base-uncased', config=BertConfig.from_pretrained('bert-base-uncased',output_hidden_states=True))