Update executor.py

2021-11-18 15:52:16 +08:00 · 2021-11-18 15:52:16 +08:00 · 24a0de00b8
commit 24a0de00b8
parent 7ba75f6fe1
1 changed files with 1 additions and 2 deletions
--- a/kws/utils/executor.py
+++ b/kws/utils/executor.py
@ -51,8 +51,7 @@ class Executor:
            if num_utts == 0:
                continue
            logits = model(feats)
-            loss, acc = criterion(logits, target, feats_lengths,
-                                         min_duration)
+            loss, acc = criterion(logits, target, feats_lengths, min_duration)
            loss.backward()
            grad_norm = clip_grad_norm_(model.parameters(), clip)
            if torch.isfinite(grad_norm):