原来的README.md改为README.old.md
- 增加
training/export_to_onnx.py,方便导出模型用于端侧部署。 - 增加
training/model_without_stft.py, 由于原来的模型中包含了关于Spectrogram的处理,用到了一些torchaudio的操作,这些操作暂时不支持直接端侧部署。我将相关处理暂时移出了模型。 - 修改
training/model_without_stft.py:line 156 - line 157, pytorch在转换torchscript时将length(int类型)处理成tensor类型导致line 159行报错,所以动态判断length的类型,如果是int类型则不转换,如果是tensor类型则转换。