无法加载模型 - Githubissues

nutalk / pytorch_bert_vec

一条命令产生bert、albert句向量，用于相似度计算和文本分类等。

34 stars 13 forks source link

无法加载模型 #3

Open howie924 opened 4 years ago

howie924 commented 4 years ago

使用的都是官方链接下载的pytorch版本的bert和albert模型（各3个文件）

transformer版本是跟要求吻合的2.4.1

运行报错： "Unable to load weights from pytorch checkpoint file. " OSError: Unable to load weights from pytorch checkpoint file. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.

nutalk commented 4 years ago

可以参考readme里面的链接下载模型。