Support exporting ONNX
## 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 条评论