lsnoo commited on
Commit
0a61521
1 Parent(s): aa8ef91

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -44
README.md CHANGED
@@ -2,47 +2,55 @@ Wav2vec2.0-xlsr-53 model is fine-tuned on commonvoice russian dataset
2
 
3
  Configs (yaml)
4
 
5
- checkpoint:
6
- save_interval: 1000 \n
7
- save_interval_updates: 1000 \n
8
- keep_interval_updates: 1
9
- no_epoch_ckechpoints: true
10
- best_checkpoint_metric: wer
11
- task:
12
- _name: audio_finetuning
13
- normalize: true
14
- labels: phn
15
- dataset:
16
- ---num_workers: 6
17
- ---max_tokens: 800000
18
- ---skip_invalid_size_inputs_valid_test: true
19
- ---valid_subset: valie
20
- -distributed_training:
21
- ---ddp_backend: legacy_ddp
22
- ---distributed_world_size: 4
23
- -criterion:
24
- ---_name: ctc
25
- ---zero_infinity: true
26
- -optimization:
27
- ---max_update: 25000
28
- ---lr: [0.00001]
29
- ---sentence_avg: true
30
- ---update_freq: [4]
31
- -optimizer:
32
- ---_name: adam
33
- ---adam_betas: (0.9, 0.98)
34
- ---adam_eps: 1e-8
35
- -lr_scheduler:
36
- ---_name: tri_stage
37
- ---phase_ratio: [0.1, 0.4, 0.5]
38
- ---final_lr_scale: 0.05
39
- -model:
40
- ---_name: wav2vec_ctc
41
- ---apply_mask: true
42
- ---mask_prob: 0.5
43
- ---mask_channel_prob: 0.1
44
- ---mask_channel_length: 64
45
- ---layerdrop: 0.1
46
- ---activation_dropout: 0.1
47
- ---feature_grad_mult: 0.0
48
- ---freeze_finetune_updates: 0
 
 
 
 
 
 
 
 
 
2
 
3
  Configs (yaml)
4
 
5
+ checkpoint:
6
+ save_interval: 1000
7
+ save_interval_updates: 1000
8
+ keep_interval_updates: 1
9
+ no_epoch_ckechpoints: true
10
+ best_checkpoint_metric: wer
11
+
12
+ task:
13
+ _name: audio_finetuning
14
+ normalize: true
15
+ labels: phn
16
+
17
+ dataset:
18
+ num_workers: 6
19
+ max_tokens: 800000
20
+ skip_invalid_size_inputs_valid_test: true
21
+ valid_subset: valie
22
+
23
+ distributed_training:
24
+ ddp_backend: legacy_ddp
25
+ distributed_world_size: 4
26
+
27
+ criterion:
28
+ _name: ctc
29
+ zero_infinity: true
30
+
31
+ optimization:
32
+ max_update: 25000
33
+ lr: [0.00001]
34
+ sentence_avg: true
35
+ update_freq: [4]
36
+
37
+ optimizer:
38
+ _name: adam
39
+ adam_betas: (0.9, 0.98)
40
+ adam_eps: 1e-8
41
+
42
+ lr_scheduler:
43
+ _name: tri_stage
44
+ phase_ratio: [0.1, 0.4, 0.5]
45
+ final_lr_scale: 0.05
46
+
47
+ model:
48
+ _name: wav2vec_ctc
49
+ apply_mask: true
50
+ mask_prob: 0.5
51
+ mask_channel_prob: 0.1
52
+ mask_channel_length: 64
53
+ layerdrop: 0.1
54
+ activation_dropout: 0.1
55
+ feature_grad_mult: 0.0
56
+ freeze_finetune_updates: 0