Unable to load weights from pytorch checkpoint file

使用pytorch加载bert预训练模型时,出现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

csdn、stackflow、github上查找了很多解决办法,但是貌似都不是我出现的问题,最后发现问题出现在下载的pytorch_model.bin文件上。

在网站https://huggingface.co/bert-base-chinese/tree/main上下载bert-base-chinese中的所有文件时,如果打开pytorch_model.bin文件,是这样子的:

这时候,如果单纯把这些内容复制到自己建的文件中,就会出现问题,所以一定要注意这个模型文件是300多M的,要点击这个按钮,

因为有一些文件,点击下载之后会出现html界面,就很容易产生先打开再把内容复制到自己建立的同名文件中的想法,注意了这一点就可以避免这种傻瓜问题的出现了!

当然,一些经典的解决办法应该还是大多数人会遇到的问题,所以打算还是给出一般的解决办法

https://blog.csdn.net/znsoft/article/details/107725285

 

你可能感兴趣的:(python,pytorch)