sarulab-speech / UTMOS22

UT-Sarulab MOS prediction system using SSL models
MIT License
169 stars 12 forks source link

How can I use pretrained model? #21

Open hoyden opened 3 months ago

hoyden commented 3 months ago

Thanks for your job! I want to evaluate your pretrained model (OOD) on my datasets. But I can't load them directly. How to use it just like in UTMOS-demo?

Takaaki-Saeki commented 3 months ago

First you may need to set up the environment as described here. Then you can download pretrained strong learners trained for OOD here. You can use these models for prediction with the command described here. If you want to run prediction with stacking, see here.

nshmyrev commented 3 months ago

There is a simple wrapper https://github.com/fakerybakery/utmos