【发布时间】:2021-08-07 22:20:56
【问题描述】:
我尝试为分类下游任务微调 BERT。
现在我再次加载模型并遇到以下警告:
初始化 BertModel 时未使用 bert-base-uncased 的模型检查点的某些权重:['cls.predictions.transform.dense.weight', 'cls.predictions.transform.LayerNorm.weight', 'cls. seq_relationship.weight','cls.predictions.bias','cls.seq_relationship.bias','cls.predictions.transform.LayerNorm.bias','cls.predictions.transform.dense.bias','cls.predictions。解码器.weight']
- 如果您是从针对其他任务或使用其他架构训练的模型的检查点初始化 BertModel(例如,从 BertForPreTraining 模型初始化 BertForSequenceClassification 模型),则这是预期的。
- 如果您从希望完全相同的模型的检查点初始化 BertModel(从 BertForSequenceClassification 模型初始化 BertForSequenceClassification 模型),则不会出现这种情况。
[屏幕截图][1] [1]:https://i.stack.imgur.com/YJZVc.png
我已经删除并重新安装了 transformers==4.6.0,但没有任何帮助。 我想也许通过参数“force_download=True”它可能会恢复原来的权重,但没有任何帮助。
我应该继续忽略警告吗?有没有办法删除模型检查点,例如在下载模型时再次固定权重?
提前致谢!
最好, 亚历克斯
【问题讨论】:
标签: nlp pytorch bert-language-model huggingface-transformers