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

阅读量：802 次

发布时间：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 exec 函数解析：探索动态代码执行的无限可能！

查看>>

Python EXEC和__NAME__

查看>>

Python exe文件打包神器-Nuitka！

查看>>

python file

查看>>

Python FileDialog获取文件夹路径不是文件

查看>>

python filter过滤器的使用_python基础知识分享：zip()、filter函数和reduce如何使用？...

查看>>

python flask 请求code 400, message Bad request version

查看>>

Python Flink Stateful函数入口上的Kafka键访问

查看>>

Python float - str - 浮动怪异

查看>>

Python Flower库：分布式任务管理与监控

python frame_python3 selenium自动化 frame表单嵌套的切换方法

查看>>

Python ftplib - 指定端口

查看>>

Python furl库：一键搞定复杂URL操作

查看>>

Python GC 也会关闭文件吗?

查看>>

python gen_key.py 报错提示找不到OpenSSL lib

查看>>

python getattrribute_Python学习——面向对象高级之反射

查看>>

Python进阶语法：字典推导式

查看>>

python gil_Python中GIL的使用详解

查看>>