Arbi-Houssem commited on
Commit
27bbe06
1 Parent(s): 20643dd

Update param_finetuninig

Browse files
Files changed (1) hide show
  1. param_finetuninig +54 -0
param_finetuninig CHANGED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project_name": "mms_tn_finetuning",
3
+ "push_to_hub": true,
4
+ "hub_model_id": "Arbi-Houssem/mms_tts_tun_Lang1.3",
5
+ "report_to": ["wandb"],
6
+ "overwrite_output_dir": true,
7
+ "output_dir": "./tmp/vits_finetuned_tn3",
8
+
9
+ "dataset_name": "Arbi-Houssem/datasetSTT-TTS",
10
+ "audio_column_name": "audio",
11
+ "text_column_name":"sentence",
12
+ "train_split_name": "train",
13
+ "eval_split_name": "validation",
14
+ "override_speaker_embeddings": true,
15
+
16
+
17
+ "full_generation_sample_text": "نحب شكون يعاوني بش نشري حاجات من عندكم",
18
+
19
+ "max_duration_in_seconds": 30,
20
+ "min_duration_in_seconds": 1.0,
21
+ "max_tokens_length": 2000,
22
+
23
+ "model_name_or_path": "Arbi-Houssem/mms_tts_tun_Lang1.3",
24
+
25
+
26
+ "preprocessing_num_workers": 4,
27
+
28
+ "do_train": true,
29
+ "num_train_epochs": 650,
30
+ "gradient_accumulation_steps": 1,
31
+ "gradient_checkpointing": true,
32
+ "per_device_train_batch_size": 8,
33
+ "learning_rate": 2e-6,
34
+ "adam_beta1": 0.8,
35
+ "adam_beta2": 0.99,
36
+ "warmup_ratio": 0.01,
37
+ "group_by_length": false,
38
+
39
+ "do_eval": true,
40
+ "eval_steps": 700,
41
+ "per_device_eval_batch_size": 8,
42
+ "max_eval_samples": 22,
43
+ "do_step_schedule_per_epoch": true,
44
+
45
+ "weight_disc": 3,
46
+ "weight_fmaps": 1,
47
+ "weight_gen": 1,
48
+ "weight_kl": 1.5,
49
+ "weight_duration": 1,
50
+ "weight_mel": 35,
51
+
52
+ "fp16": true,
53
+ "seed": 456
54
+ }