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......

阅读量：800 次

发布时间：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 bool() 函数能否为无效参数引发异常?

查看>>

Python C 程序子进程在“for line in iter“处挂起

查看>>

Python Celery：自动化测试平台定时任务必备的三方库

查看>>

python check_output 失败，退出状态为 1，但 Popen 适用于相同的命令

查看>>

Python CONNECT 4 CHECK WIN函数

查看>>

python cos,Python cos（90）和cos（270）不是0

查看>>

python进阶（4）：Python 脚本文件重启自身进程

查看>>

python ctypes库中动态链接库加载方式

查看>>

python cv2 图像( np array )转 HObject

python进阶（2）：pyecharts使用

查看>>

python cv2读取rtsp实时码流按时生成连续视频文件

查看>>

Python Dataframe Groupby Mean和Std

Python Day17 Django 03

查看>>