minor fixes to the matcha recipe

k2-fsa · Dec 9, 2024 · 08caa1e · 08caa1e
1 parent 32b7a44
commit 08caa1e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/egs/ljspeech/TTS/matcha/train.py b/egs/ljspeech/TTS/matcha/train.py
@@ -488,9 +488,10 @@ def save_bad_model(suffix: str = ""):
 
                 loss = sum(losses.values())
 
-                optimizer.zero_grad()
                 scaler.scale(loss).backward()
                 scaler.step(optimizer)
+                scaler.update()
+                optimizer.zero_grad()
 
                 loss_info = MetricsTracker()
                 loss_info["samples"] = batch_size