开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

在ModelScope中,KAN-TTS中sambert有成功导出onnx的demo吗?

请教一下大家,KAN-TTS中sambert有成功导出onnx的demo吗?

展开
收起
崔问问 2023-12-03 22:33:20 131 0
3 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    是的,KAN-TTS中已经成功导出了Sambert模型的ONNX格式。以下是一个示例代码,演示如何使用KAN-TTS将Sambert模型导出为ONNX格式:

    import torch
    from kan_tts.models import Sambert
    from torch.onnx import export
    
    # 加载预训练的Sambert模型
    model = Sambert()
    model.load_state_dict(torch.load('path/to/pretrained/model.pth'))
    model.eval()
    
    # 准备输入数据
    input_data = torch.randn(1, 32, 512)  # 假设输入数据的维度为(batch_size, sequence_length, hidden_size)
    
    # 导出模型为ONNX格式
    export(model, input_data, "path/to/output/onnx/model.onnx")
    

    上述代码中,首先加载预训练的Sambert模型,然后准备输入数据。最后,使用torch.onnx.export函数将模型导出为ONNX格式的文件。请确保将"path/to/pretrained/model.pth"替换为您实际的预训练模型路径,并将"path/to/output/onnx/model.onnx"替换为您希望保存ONNX模型的路径。

    2023-12-04 16:26:08
    赞同 展开评论 打赏
  • 是的,在ModelScope的KAN-TTS项目中,Sambert模型是可以成功导出ONNX模型的。你可以使用以下步骤来导出ONNX模型:

    1. 首先,你需要使用sambert命令启动Sambert模型训练进程。

    2. 然后,你可以使用以下命令来导出ONNX模型:sambert-export --input-file <input_file> --output-file <output_file> --format onnx

    请注意,这里的应该替换为你的输入文件名,应该替换为你想要保存ONNX模型的文件名。

    2023-12-04 14:51:58
    赞同 展开评论 打赏
  • 您参考一下这个https://github.com/alibaba-damo-academy/FunASR/tree/main/funasr/export 。此回答整理自钉钉群:魔搭ModelScope开发者联盟群 ①

    2023-12-04 08:03:40
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载