| 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) | |