python bert_gen.py 报错Unable to load weights from pytorch checkpoint file for......-白红宇

python bert_gen.py 报错Unable to load weights from pytorch checkpoint file for......

阅读量：806 次

发布时间：2023-03-05

本文共 613 字，大约阅读时间需要 2 分钟。

环境：运行环境：WIN10BERT版本：Bert-VITS2.4

问题描述：运行Python脚本bert_gen.py时，出现以下错误：

Unable to load weights from pytorch checkpoint file for './bert/Erlangshen-MegatronBert-1.3B-Chinese/pytorch_model.bin'.If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.

此错误提示表明尝试从PyTorch模型中加载TensorFlow 2.0的检查点文件。

解决方案：

确认模型文件路径是否正确

检查PyTorch和TensorFlow的版本兼容性

如果是从TensorFlow检查点加载，请在初始化模型时设置from_tf=True

具体操作步骤：

确认模型文件路径是否正确，确保文件路径无误

如果模型文件大小不符（约14.8GB），请重新下载完整文件

确保PyTorch和TensorFlow的版本兼容

在加载模型时，检查是否需要设置from_tf=True（如从TensorFlow检查点加载）

如果问题依旧，建议清除旧的PyTorch检查点文件并重新初始化模型

经过上述步骤后，应能够正常运行python bert_gen.py脚本。

转载地址：http://pvafk.baihongyu.com/

你可能感兴趣的文章

python编程基础之十六

查看>>

python字符串input输入_Python input()函数：获取用户输入的字符串

查看>>

python 对json数据读取及保存与读取，对dump，dumps，load，loads的理解

Python 导入requests报错No module named requests

查看>>

Python 将JPEG图片批量改成jpg并删除JPEG图片