博客
关于我
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可视化matplotlib_如何使用Python中最强大的可视化工具Matplotlib?
    查看>>
    python可维护性_使用这7大神器,让你的Python 代码更容易于维护
    查看>>
    Python只运行一次while循环
    查看>>
    python变量的详细教程_Python零基础入门教程之语法入门[变量](第二期)
    查看>>
    Python变量命名方法-ChatGPT4o作答
    查看>>
    Python变量与运算符
    查看>>
    Python变量/运算符/函数/模块/string
    查看>>
    python发送邮件的时候出现 error (535, b‘5.7.3 Authentication unsuccessful‘) 解决方法
    查看>>
    python系列【仅供参考】:python flask框架 debug功能
    查看>>
    python发送notes邮件_使用python在Lotus Notes中发送邮件
    查看>>
    Python双版本下创建一个Scrapy(西瓜皮)项目
    查看>>
    Python双版本下No module named 'requests'
    查看>>
    python及pycharm2018软件安装教程
    查看>>
    python去重txt文本_Python实现的txt文件去重功能示例
    查看>>
    python去掉列表的逗号,从Python列表项中删除标点符号
    查看>>
    Python卸载所有包
    查看>>
    python单线程下实现多个socket并发
    查看>>
    Python单元测试框架介绍(超详细~)
    查看>>
    Python单元测试框架
    查看>>