Closed axuan731 closed 10 months ago
可以参考run.sh里这一步,convert_model.pt是已经重参数化之后的直筒型结构,直接用该模型导出onnx即可
可以参考run.sh里这一步,convert_model.pt是已经重参数化之后的直筒型结构,直接用该模型导出onnx即可
感谢您的回复,我没有表达清楚,我已经转换成了直筒型结构的模型,导出onnx需要直筒型模型的网络,但是repvgg.py不是直筒型的网络。
可以参考run.sh里这一步,convert_model.pt是已经重参数化之后的直筒型结构,直接用该模型导出onnx即可
感谢您的回复,我没有表达清楚,我已经转换成了直筒型结构的模型,导出onnx需要直筒型模型的网络,但是repvgg.py不是直筒型的网络。
只需要在配置文件中加入configs['model_args']['deploy'] = True
参数,就能得到直筒型的结构。
run.sh中调用wespeaker/models/convert_repvgg.py
导出的直筒型结构模型,并且在配置文件中写入configs['model_args']['deploy'] = True
这个参数,详细如下:
https://github.com/wenet-e2e/wespeaker/blob/c9ec537b53fe1e04525be74b2550ee95bed3a891/wespeaker/models/convert_repvgg.py#L25-L34
非常感谢~
您好: 我发现RepVGG由于结构重参数化与其他pt模型转onnx不太一样, 请问如何实现RepVGG结构重参数化之后转onnx? 谢谢