ITADN

Support exporting ONNX

#1Pull Requestyqzhishen 创建于 2023-07-10已合并
Y
yqzhishencommented
## Requirements - PyTorch nightly (>=2.1.0), for exporting aten::stft to ONNX opset version 17 - Latest version of `onnx` and `onnxsim` packages, for ONNX graph optimization ## Usage ```bash python export.py -m path/to/your/model.pt -o path/to/your/model.onnx --optimize ``` - ONNX inputs: waveform, threshold - ONNX outputs: f0, uv ## Known Issues - `hop_length` must be static (specified by the `--hop_length` CLI option) due to PyTorch bug.
合并状态:已合并 合并于 2023-07-17 关闭于 2023-07-17 0 条评论