unilight commited on
Commit
967db88
1 Parent(s): 3a2565f

Upload nisqa/sslmos/4337/config.yml with huggingface_hub

Browse files
Files changed (1) hide show
  1. nisqa/sslmos/4337/config.yml +58 -0
nisqa/sslmos/4337/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-4337
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: 4337
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