rust/sbv2-api
1
0
Fork 0
mirror of https://github.com/neodyland/sbv2-api.git synced 2026-01-08 07:22:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: build optimization, convert model onnx

Browse Source
This commit is contained in:
Googlefan
2024-09-22 06:00:57 +00:00
parent 290fb37c16
commit 962fa9a49d
3 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
convert/convert_model.py
View File

@@ -126,11 +126,13 @@ torch.onnx.export(
f"../models/model_{out_name}.onnx",
verbose=True,
dynamic_axes={
"x_tst": {1: "batch_size"},
"x_tst": {0: "batch_size", 1: "x_tst_max_length"},
"x_tst_lengths": {0: "batch_size"},
"tones": {1: "batch_size"},
"language": {1: "batch_size"},
"bert": {2: "batch_size"},
"sid": {0: "batch_size"},
"tones": {0: "batch_size", 1: "x_tst_max_length"},
"language": {0: "batch_size", 1: "x_tst_max_length"},
"bert": {0: "batch_size", 2: "x_tst_max_length"},
"style_vec": {0: "batch_size"},
},
input_names=[
"x_tst",