From 3d13bd7264986a2899f33c890ad8e2a03c94ad5e Mon Sep 17 00:00:00 2001
From: "Aaron (Yinghao) Li" <71044569+yl4579@users.noreply.github.com>
Date: Tue, 24 Jan 2023 13:20:50 -0500
Subject: [PATCH] Update config.yml

---
 Configs/config.yml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/Configs/config.yml b/Configs/config.yml
index 03fdda8..3dad1af 100644
--- a/Configs/config.yml
+++ b/Configs/config.yml
@@ -45,6 +45,9 @@ loss_params:
     lambda_s2s: 1. # sequence-to-sequence loss (1st stage, TMA)
     TMA_epoch: 20 # TMA starting epoch (1st stage)
 
+    # https://github.com/yl4579/StyleTTS/issues/7
+    TMA_CEloss: false # whether to use cross-entropy (CE) loss for TMA
+
     lambda_F0: 1. # F0 reconstruction loss (2nd stage)
     lambda_norm: 1. # norm reconstruction loss (2nd stage)
     lambda_dur: 1. # duration loss (2nd stage)