tensorflow加载预训练模型报错:
tensorflow.python.framework.errors_impl.DataLossError: Unable to open table file /home/chinese_L-12_H-768_A-12/bert_model.ckpt.data-00000-of-00001: Data loss: not an sstable (bad magic number): perhaps your file is in a different file format and you need to use a different restore operator?
完整报错如下图:
很可能是因为预训练模型的路径没有写对,如下我的预训练模型路径为:
我的路径为:
/home/chinese_L-12_H-768_A-12/bert_model.ckpt.data-00000-of-00001
实际路径应该为:
/home/chinese_L-12_H-768_A-12/bert_model.ckpt
即还应该包含bert_model.ckpt.index,bert_model.ckpt.meta这两个文件!
Done!!!