File size: 628 Bytes
aa29130
 
1ae875b
 
8c93826
93097a6
1ae875b
 
aa29130
 
 
 
 
 
 
 
1ae875b
aa29130
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from espnet2.bin.tts_train import train
from espnet2.tasks.tts import TTSTask
from datasets import load_dataset

# تحميل البيانات للهجة النجدية
dataset = load_dataset("m6011/sada2022")
najdi_data = dataset.filter(lambda example: example['SpeakerDialect'] == 'Najdi')

# إعداد التدريب
train_config = {
    'output_dir': './results',
    'train_data_path_and_name_and_type': najdi_data['train'],
    'valid_data_path_and_name_and_type': najdi_data['test'],
    'train_batch_size': 8,
    'epochs': 10,
}

# بدء عملية التدريب باستخدام ESPnet
TTSTask.main(**train_config)