博客
关于我
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 pandas TimeStamps到夏令时的本地时间字符串
    查看>>
    Python Pandas 用顶行替换标题
    查看>>
    Python Pandas-从DataFrame按类别绘制多个条形图
    查看>>
    Python PyQt5 将不再显示此消息复选框添加到 QMessageBox
    查看>>
    Python PyQt5:如何使用 PyQt5 显示错误消息
    查看>>
    Python pytest 面试题!
    查看>>
    Python pytz 时区函数返回一个相差 9 分钟的时区
    查看>>
    python rabbitmq实现简单/持久/广播/组播/topic/rpc消息异步发送可配置Django
    查看>>
    Python random和json模块
    查看>>
    Python random模块seed理解
    查看>>
    python range()函数
    查看>>
    Python rdflib可传递查询
    查看>>
    python redis 集群_python 搭建redis集群
    查看>>
    python redis连接,在Python中使用Redis连接池的正确方法
    查看>>
    python regex_Python RegEx
    查看>>
    python requests post 中文结果请求得到unicode
    查看>>
    Python Requests接口自动化测试实战
    查看>>
    Python requests模块
    查看>>
    python request与grequests该如何选择
    查看>>
    python request模块
    查看>>