sambert采用torch.onnx.export()时输出一堆warning,但并没有报错,就一直卡在一个warning之后。
一直卡在最后这个warning处,一晚上都没动。torch执行推理的时候正常,可以完全正常执行完forward,但就是导出onnx不行,求解?
要将语音合成模型 SamBERT 和 HiFi-GAN 转换为 ONNX 模型,您可以按照以下步骤进行操作:
torch.onnx.export(hifigan_model, input_audio, "hifigan_model.onnx")
```
请注意,上述代码中的 sambert_model 和 hifigan_model 是您已经训练好的 SamBERT 和 HiFi-GAN 模型的实例,input_audio 是用于导出模型的输入音频数据或文件。
关于您的 SamBERT 导出失败的问题,您可以尝试以下方法来解决:
希望这些信息能够帮助您将 SamBERT 和 HiFi-GAN 转换为 ONNX 模型。如果您有任何其他问题,请随时提问。
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。