unilight commited on
Commit
3254318
·
verified ·
1 Parent(s): dd1382a

Upload bvcc+somos+singmos+nisqa+tmhint-qi+tencent+pstn+urgent2024-mos/sslmos-wavlm_large/1337/config.yml with huggingface_hub

Browse files
bvcc+somos+singmos+nisqa+tmhint-qi+tencent+pstn+urgent2024-mos/sslmos-wavlm_large/1337/config.yml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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-wavlm_large.yaml
8
+ dataset_type: NonIntrusiveDataset
9
+ dev_csv_path: data/dev.csv
10
+ dev_samples_per_eval_loop: 1000
11
+ distributed: false
12
+ eval_and_save_interval_steps: 1000
13
+ grad_norm: 1.0
14
+ gradient_accumulate_steps: 2
15
+ inference_mode: mean_net
16
+ init_checkpoint: ''
17
+ keep_nbest_models: 5
18
+ listener_score_criterions: null
19
+ load_wav_cache: true
20
+ log_interval_steps: 100
21
+ mean_score_criterions:
22
+ - criterion_params:
23
+ masked_loss: false
24
+ order: 1
25
+ tau: 0.5
26
+ criterion_type: ScalarLoss
27
+ criterion_weight: 1.0
28
+ model_input: waveform
29
+ model_params:
30
+ mean_net_dnn_dim: 64
31
+ mean_net_output_type: scalar
32
+ mean_net_range_clipping: true
33
+ s3prl_name: wavlm_large
34
+ ssl_model_layer_idx: -1
35
+ ssl_model_output_dim: 1024
36
+ ssl_module: s3prl
37
+ use_listener_modeling: false
38
+ use_mean_listener: false
39
+ model_type: SSLMOS
40
+ num_domains: 8
41
+ num_workers: 0
42
+ optimizer_params:
43
+ lr: 0.001
44
+ momentum: 0.9
45
+ optimizer_type: SGD
46
+ outdir: exp/ssl-mos-wavlm_large-1337
47
+ padding_mode: repetitive
48
+ patience: 20
49
+ pin_memory: true
50
+ rank: 0
51
+ resume: ''
52
+ sampling_rate: 16000
53
+ scheduler_type: null
54
+ seed: 1337
55
+ test_batch_size: 1
56
+ train_batch_size: 8
57
+ train_csv_path: data/train.csv
58
+ train_max_steps: 150000
59
+ trainer_type: NonIntrusiveEstimatorTrainer
60
+ verbose: 1
61
+ version: 0.1.0
62
+ wav_only: true