espnet / espnet_onnx

Onnx wrapper for espnet infrernce model
MIT License
152 stars 24 forks source link

Bugfix/fix stft frontend #71

Closed Masao-Someki closed 1 year ago

Masao-Someki commented 1 year ago

The current implementation contains a padding issue with librosa.stft. As a result, the output of torch.stft and librosa.stft is different, which leads to accuracy degradation. In this PR, I fixed that padding mode from 'constant,' which is the default HP of librosa.stft, into 'reflect,' which is the default HP of torch.stft.