macanv / BERT-BiLSTM-CRF-NER

Tensorflow solution of NER task Using BiLSTM-CRF model with Google BERT Fine-tuning And private Server services
https://github.com/macanv/BERT-BiLSMT-CRF-NER
4.67k stars 1.25k forks source link

adam_filter后的问题 #388

Open sly123197811 opened 3 years ago

sly123197811 commented 3 years ago

模型训练结束后,我使用提供的adam_filter方法,删除一些不必要的参数后,重新加载模型做预测,报以下错误: tensorflow.python.framework.errors_impl.NotFoundError: Restoring from checkpoint failed. This is most likely due to a Variable name or other graph key that is missing from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error: Key project/hidden/W not found in checkpoint

我打印bert和新模型所有参数后,发现并没有project/hidden/W,希望楼主指点一下,万分感谢