k2-fsa
diff --git a/‎egs/ljspeech/TTS/vits/train.py
+1 b/‎egs/ljspeech/TTS/vits/train.py
+1
diff --git a/‎egs/ljspeech/TTS/vits2/flow.py
-311 b/‎egs/ljspeech/TTS/vits2/flow.py
-311
diff --git a/‎egs/ljspeech/TTS/vits2/flow.py
+1 b/‎egs/ljspeech/TTS/vits2/flow.py
+1
@@ -405,6 +405,7 @@ def save_bad_model(suffix: str = ""):
                 )
             for k, v in stats_d.items():
                 loss_info[k] = v * batch_size
+
             # update discriminator
             optimizer_d.zero_grad()
             scaler.scale(loss_d).backward()
 
@@ -0,0 +1 @@
+../../../ljspeech/TTS/vits/flow.py
Original file line number	Diff line number	Diff line change
`@@ -405,6 +405,7 @@ def save_bad_model(suffix: str = ""):`
`405`	`405`	`)`
`406`	`406`	`for k, v in stats_d.items():`
`407`	`407`	`loss_info[k] = v * batch_size`
	`408`	`+`
`408`	`409`	`# update discriminator`
`409`	`410`	`optimizer_d.zero_grad()`
`410`	`411`	`scaler.scale(loss_d).backward()`