Adding the possibility to train with durations

2026-02-04 17:59:19 +08:00 · 2024-05-27 13:24:21 +02:00
parent e658aee6a5
commit aa496aa13f
7 changed files with 54 additions and 12 deletions
--- a/matcha/models/baselightningmodule.py
+++ b/matcha/models/baselightningmodule.py
@@ -65,6 +65,7 @@ class BaseLightningClass(LightningModule, ABC):
            y_lengths=y_lengths,
            spks=spks,
            out_size=self.out_size,
+            durations=batch["durations"],
        )
        return {
            "dur_loss": dur_loss,