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.