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

阅读量：798 次

发布时间：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/

你可能感兴趣的文章

Prometheus 云原生 - Prometheus 数据模型、Metrics 指标类型、Exporter 相关

查看>>

Prometheus 云原生 - 基于 file_sd、http_sd 实现 Service Discovery

查看>>

Prometheus 云原生 - 微服务监控报警系统（Promethus、Grafana、Node_Exporter）部署、简单使用

prometheus 安装node_exporter， node_exporter 安装最新版普罗米修思安装监控服务器client

Prometheus监控redis数据库实战

查看>>

Prometheus监控教程：使用Grafana展示主机基本信息

查看>>

Prometheus监控教程：配置介绍

查看>>

Prometheus（2）：SpringBoot 2.X集成Prometheus

查看>>

Promise 原理解析与实现(遵循Promise/A+规范)

查看>>

PyTorch:传递 numpy 数组进行权重初始化

查看>>

promise.all是并发执行吗_攻破面试灵魂拷问，解读Java并发编程的艺术，本文带你深入l理解...

查看>>

PyTorch-Tutorials【pytorch官方教程中英文详解】- 7 Optimization

查看>>

promise总结

查看>>