{"repo":"Blinorot/utmos-pytorch","free":true,"listed":false,"github":"https://github.com/Blinorot/utmos-pytorch","clone":"git clone https://github.com/Blinorot/utmos-pytorch.git","description":"Unofficial fairseq-free PyTorch implementation of UTMOS (v1, 2022), matching the original system.","language":"Python","stars":35,"topics":["metrics","mos","mos-prediction","neural-audio-codec","pytorch","text-to-speech","torchscript","utmos","audio-quality-assessment"],"license":"MIT","category":"machine-learning","readme_excerpt":"UTMOS-PyTorch About • Usage • How To Reproduce • Credits • License • Citation About This is an unofficial fairseq -free implementation of the UTMOS MOS Prediction system proposed in UTMOS: UTokyo-SaruLab System for VoiceMOS Challenge 2022. The original implementation is based on fairseq. However, fairseq is difficult to install with recent Python, PyTorch, and dependency versions, which makes UTMOS hard to use in modern environments. Recent study from ICASSP 2026 highlights the high correlation of UTMOS with subjective listening scores for neural codecs. Therefore, modern neural audio codec and TTS research benefits from an easy-to-install UTMOS implementation. We provide a fairseq -free implementation written in PyTorch that matches the original system using converted weights and re-written modules. We also provide a TorchScript variant that can be loaded with only PyTorch, without installing this package. The PyTorch and TorchScript versions are validated against the original implementation and produce matching scores. [!NOTE] As in the original version, we recommend running UTMOS with batch size 1 to avoid metric shifts caused by padding. Usage You can install the repo as a package: Or from source: The code requires: Package Version --------------- ------- Python =3.9 PyTorch =2.2.0 HuggingFace Hub =0.20 The TorchScript checkpoint was scripted with PyTorch 2.5.1 . We have tested that it works on PyTorch 2.2.0 , however, PyTorch =2.5.1 is recommended for the TorchScript var","default_branch":null,"files":null,"tree":[],"storefront":"/r/Blinorot","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/Blinorot/utmos-pytorch/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}