search

sarulab-speech / UTMOS22

UT-Sarulab MOS prediction system using SSL models
MIT License
148 stars 14 forks source link

Change SSL model #13

Closed coding-amateur1415 closed 10 months ago

coding-amateur1415 commented 1 year ago

Hello, I have observed that you are utilizing wav2vec_small.pt as your SSL model. what are the steps to configure a different SSL model like wav2vec(large)?

Takaaki-Saeki commented 10 months ago

Hi, sorry for the delay. To change the SSL model,

  1. Please run download_stacking_checkpoints.sh to download different SSL models.
  2. Specify the SSL model path in https://github.com/sarulab-speech/UTMOS22/blob/6f57475688912d11104fefcb43dde18728f3cc68/strong/configs/model/default.yaml#L9, which is used here. Then run a training job.