博客
关于我
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列表中删除重复元素,同时保留顺序?...
    查看>>
    python列表前几个_python之列表
    查看>>
    python列表元组
    查看>>
    Python列表/元组/字典和集合使用
    查看>>
    python列表 行列选择_python_pandas_dataframe_行列选择_切片操作
    查看>>
    python列表
    查看>>
    python列出当前目录、子目录和文件的脚本
    查看>>
    python+flask计算机毕业设计骨科门诊患者档案管理系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校体测管理系统的设计与实现(程序+开题+论文)
    查看>>
    Python切片对象和__getitem__
    查看>>
    python+flask计算机毕业设计高校体育竞赛成绩管理系统的设计与实现(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校体育赛事系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校体育运动会比赛系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校危化品管理系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校团委事务管理系统(程序+开题+论文)
    查看>>
    Python切片(%s)
    查看>>
    python+flask计算机毕业设计高校学生选课系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校学生饮食推荐系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校教务管理系统(程序+开题+论文)
    查看>>
    python+flask计算机毕业设计高校教学档案管理系统(程序+开题+论文)
    查看>>