BERT实体识别转换PB文件采坑记录

BERT实体识别转换PB文件采坑记录

---

前言

记录一下转换ckpt到pb文件的采坑记录

---

采坑代码为https://github.com/macanv/BERT-BiLSTM-CRF-NER，忍不住说脏话了，好家伙这位大佬的代码用一次坑一次，你永远不知道下一次采坑会在何处。

一、遇到的错误

本来是想着还是使用https://github.com/xmxoxo/BERT-train2deploy这里的freeze.py文件完成转换，但是运行之后发现一只存在一个bug，如下图。看了网上的许多解决办法，也看了原作者的GitHub，发现如果不是用分类数据，好像都会出现这个问题。鉴于这位大佬的程序也是从macanv中改善的，我还是决定用源码试一下。但是我2022年重新做的时候发现我用的冻结参数的程序是它本身自带的graph程序，但是我忘记了当时这么做的原因是什么。。。

二、macanv代码的采坑记录

1.以前采坑

这位大佬的代码我不能说一步一个坑，我只能说我自己太菜鸡，这个readme和没有一样，这次也是，刚开始输入的运行命令如下：

python /data/lxs/definition-NER-bilstm/bert_base