Skip to content

csu-eis/sota-music-tagging-models-for-inference

 
 

Repository files navigation

README

原来的README.md改为README.old.md

修改内容

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

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%