HopeLiang commited on
Commit
526e6b7
·
verified ·
1 Parent(s): 5aac3ff

Add model card

Browse files
Files changed (1) hide show
  1. README.md +49 -0
README.md ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - speech
4
+ - speech-quality-assessment
5
+ - mos-prediction
6
+ - pytorch
7
+ ---
8
+
9
+ # SSL Layer MOS checkpoints
10
+
11
+ This repository contains the projection-head checkpoints associated with
12
+ [SSL_Layer_MOS](https://github.com/Hope-Liang/SSL_Layer_MOS), the official
13
+ implementation for *Selection of Layers from Self-supervised Learning Models
14
+ for Predicting Mean-Opinion-Score of Speech* (IEEE ASRU 2025).
15
+
16
+ ## Contents
17
+
18
+ The checkpoints are grouped by evaluation dataset:
19
+
20
+ - `results_nisqa/`
21
+ - `results_tencent/`
22
+ - `results_bvcc/`
23
+
24
+ Within each dataset, files are organized by SSL model, layer, and run. Each
25
+ experiment includes PyTorch checkpoint files (`.pt`) and its Gin configuration
26
+ (`.gin`); logs are retained for reproducibility.
27
+
28
+ ## Usage
29
+
30
+ Clone the companion code repository and follow its inference instructions:
31
+
32
+ ```bash
33
+ git clone https://github.com/Hope-Liang/SSL_Layer_MOS.git
34
+ cd SSL_Layer_MOS
35
+ python inference.py --audio_path=<YOUR_AUDIO_PATH> --ssl_model=<SELECTED_SSL_MODEL> --ssl_layer=<SELECTED_SSL_LAYER> --ckpt=<PATH_TO_DOWNLOADED_CHECKPOINT>
36
+ ```
37
+
38
+ Download a checkpoint from the directory matching your selected dataset, SSL
39
+ model, layer, and run.
40
+
41
+ ## Citation
42
+
43
+ If you use these checkpoints, please cite the associated IEEE ASRU 2025 paper
44
+ and the [companion implementation](https://github.com/Hope-Liang/SSL_Layer_MOS).
45
+
46
+ ## License
47
+
48
+ The license for these checkpoints follows the terms specified by the authors.
49
+ Please also comply with the licenses of any underlying SSL models and datasets.