tflite 模型转换出现问题tflite Check failed: axis < input_shape.dimensions count()

[LLLLLLoki]

感谢您的工作，我在使用您的代码做NER 时出现了一下问题： 2020-08-03 19:11:21.629155: F tensorflow/lite/toco/graph_transformations/propagate_fixed_sizes.cc:1813] Check failed: axis < input_shape.dimensions_count() (62 vs. 3) Fatal Python error: Aborted

Current thread 0x00007f0435b05700 (most recent call first): File "/home/lxj/anaconda3/envs/nlp-gpu1.14/lib/python3.7/site-packages/tensorflow/lite/toco/python/toco_from_protos.py", line 33 in execute File "/home/lxj/anaconda3/envs/nlp-gpu1.14/lib/python3.7/site-packages/absl/app.py", line 250 in _run_main File "/home/lxj/anaconda3/envs/nlp-gpu1.14/lib/python3.7/site-packages/absl/app.py", line 299 in run File "/home/lxj/anaconda3/envs/nlp-gpu1.14/lib/python3.7/site-packages/tensorflow/python/platform/app.py", line 40 in run File "/home/lxj/anaconda3/envs/nlp-gpu1.14/lib/python3.7/site-packages/tensorflow/lite/toco/python/toco_from_protos.py", line 59 in main File "/home/lxj/anaconda3/envs/nlp-gpu1.14/bin/toco_from_protos", line 11 in Aborted (core dumped) 我在Albert 后面直接使用softmax 进行预测，理论上与原模型相似的，但是为什么却不能转换成TFlite