博客
关于我
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 列表(List)概述-ChatGPT4o作答
    查看>>
    Python 初学者需要知道的四条建议
    查看>>
    Python 判断字符串是否包含子字符串
    查看>>
    python 判断当前时间是否为零点
    查看>>
    python 利用pandas读取本地中CSV文件的指定列 列名重命名 并保存回本地
    查看>>
    python 利用pyspark读取HDFS中CSV文件的指定列 列名重命名 并保存回HDFS
    查看>>
    python 利用pyttsx3文字转语音
    查看>>
    python 利用已有Ner模型进行数据清洗合并
    查看>>
    python 到大数据开发工程师_如何成为一个大数据开发工程师?
    查看>>
    python 加密解密(base64, AES)
    查看>>
    Python 和Java 哪个更适合做自动化测试?
    查看>>
    python 图片转ico
    查看>>
    python 图片转文字、语音转文字、文字转语音保存音频并朗读
    查看>>
    python 在包含类似字符\x16、\x12、\x某某的数组中将以\x开头的字符找出来的方法
    查看>>
    Python 在并行进程之间共享字典
    查看>>
    Python 垃圾收集器文档
    查看>>
    python 基于 wordcloud + jieba + matplotlib 生成词云
    查看>>
    python 基于detectron或mask_rcnn的mask遮罩区域进行图片截取
    查看>>