From 29e166063e442648b12e9e5318efa2caeb7e9fb7 Mon Sep 17 00:00:00 2001
From: Ethan <45615462+EthanLifeGreat@users.noreply.github.com>
Date: Tue, 26 Nov 2024 10:32:14 +0800
Subject: [PATCH] Update tts_aept.v1.yaml

Add `trainer_type` and `collater_type` assignments  to prevent using the default `ARTTSTrainer` or `ARTTSCollator`.
---
 egs/ljspeech/tts1/conf/tts_aept.v1.yaml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/egs/ljspeech/tts1/conf/tts_aept.v1.yaml b/egs/ljspeech/tts1/conf/tts_aept.v1.yaml
index 2c43a6f..9a22977 100644
--- a/egs/ljspeech/tts1/conf/tts_aept.v1.yaml
+++ b/egs/ljspeech/tts1/conf/tts_aept.v1.yaml
@@ -30,6 +30,12 @@ model_params:
     decoder_concat_after: False
     decoder_reduction_factor: 2
 
+###########################################################
+#                      LOSS SETTING                       #
+###########################################################
+trainer_type: ARVCTrainer
+collater_type: ARVCCollater
+
 ###########################################################
 #                   INFERENCE SETTING                     #
 ###########################################################
@@ -46,4 +52,4 @@ batch_size: 100             # Batch size.
 ###########################################################
 #                    INTERVAL SETTING                     #
 ###########################################################
-train_max_steps: 50000                 # Number of training steps.
\ No newline at end of file
+train_max_steps: 50000                 # Number of training steps.