Upload nisqa/sslmos/3337/config.yml with huggingface_hub
Browse files- nisqa/sslmos/3337/config.yml +58 -0
nisqa/sslmos/3337/config.yml
ADDED
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
additional_config: null
|
2 |
+
allow_cache: true
|
3 |
+
best_model_criterion:
|
4 |
+
key: utt_SRCC
|
5 |
+
order: highest
|
6 |
+
collater_type: NonIntrusiveCollater
|
7 |
+
config: conf/ssl-mos-wav2vec2.yaml
|
8 |
+
dataset_type: NonIntrusiveDataset
|
9 |
+
dev_csv_path: data/nisqa_dev.csv
|
10 |
+
distributed: false
|
11 |
+
eval_and_save_interval_steps: 100
|
12 |
+
grad_norm: 1.0
|
13 |
+
inference_mode: mean_net
|
14 |
+
init_checkpoint: ''
|
15 |
+
keep_nbest_models: 5
|
16 |
+
listener_score_criterions: null
|
17 |
+
log_interval_steps: 100
|
18 |
+
mean_score_criterions:
|
19 |
+
- criterion_params:
|
20 |
+
masked_loss: false
|
21 |
+
order: 1
|
22 |
+
tau: 0.5
|
23 |
+
criterion_type: ScalarLoss
|
24 |
+
criterion_weight: 1.0
|
25 |
+
model_input: waveform
|
26 |
+
model_params:
|
27 |
+
mean_net_dnn_dim: 64
|
28 |
+
mean_net_output_type: scalar
|
29 |
+
mean_net_range_clipping: true
|
30 |
+
s3prl_name: wav2vec2
|
31 |
+
ssl_model_layer_idx: -1
|
32 |
+
ssl_model_output_dim: 768
|
33 |
+
ssl_module: s3prl
|
34 |
+
use_listener_modeling: false
|
35 |
+
use_mean_listener: false
|
36 |
+
model_type: SSLMOS
|
37 |
+
num_workers: 0
|
38 |
+
optimizer_params:
|
39 |
+
lr: 0.001
|
40 |
+
momentum: 0.9
|
41 |
+
optimizer_type: SGD
|
42 |
+
outdir: exp/ssl-mos-wav2vec2-3337
|
43 |
+
padding_mode: repetitive
|
44 |
+
patience: 20
|
45 |
+
pin_memory: true
|
46 |
+
rank: 0
|
47 |
+
resume: ''
|
48 |
+
sampling_rate: 16000
|
49 |
+
scheduler_type: null
|
50 |
+
seed: 3337
|
51 |
+
test_batch_size: 1
|
52 |
+
train_batch_size: 16
|
53 |
+
train_csv_path: data/nisqa_train.csv
|
54 |
+
train_max_steps: 100000
|
55 |
+
trainer_type: NonIntrusiveEstimatorTrainer
|
56 |
+
verbose: 1
|
57 |
+
version: 0.1.0
|
58 |
+
wav_only: true
|