From af12925142cd506c25c22b9cfb10be9726f2a33d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Mon, 21 Jun 2021 17:23:24 +0200
Subject: [PATCH] =?UTF-8?q?Add=20AlignTTS=20recipe=20=F0=9F=91=A9=E2=80=8D?=
 =?UTF-8?q?=F0=9F=8D=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 recipes/ljspeech/align_tts/train_aligntts.py | 30 ++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 recipes/ljspeech/align_tts/train_aligntts.py

diff --git a/recipes/ljspeech/align_tts/train_aligntts.py b/recipes/ljspeech/align_tts/train_aligntts.py
new file mode 100644
index 00000000..4a4f86c4
--- /dev/null
+++ b/recipes/ljspeech/align_tts/train_aligntts.py
@@ -0,0 +1,30 @@
+import os
+
+from TTS.tts.configs import AlignTTSConfig
+from TTS.tts.configs import BaseDatasetConfig
+from TTS.trainer import init_training, Trainer, TrainingArgs
+
+
+output_path = os.path.dirname(os.path.abspath(__file__))
+dataset_config = BaseDatasetConfig(name="ljspeech", meta_file_train="metadata.csv", path=os.path.join(output_path, "../LJSpeech-1.1/"))
+config = AlignTTSConfig(
+    batch_size=32,
+    eval_batch_size=16,
+    num_loader_workers=4,
+    num_eval_loader_workers=4,
+    run_eval=True,
+    test_delay_epochs=-1,
+    epochs=1000,
+    text_cleaner="english_cleaners",
+    use_phonemes=False,
+    phoneme_language="en-us",
+    phoneme_cache_path=os.path.join(output_path, "phoneme_cache"),
+    print_step=25,
+    print_eval=True,
+    mixed_precision=False,
+    output_path=output_path,
+    datasets=[dataset_config]
+)
+args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config)
+trainer = Trainer(args, config, output_path, c_logger, tb_logger)
+trainer.fit()