fixed the parameter transfer problem for criterion

2022-07-13 21:42:52 +08:00 · 2022-07-13 21:42:52 +08:00 · 9003703d6c
commit 9003703d6c
parent 141d40704f
1 changed files with 1 additions and 1 deletions
--- a/kws/utils/executor.py
+++ b/kws/utils/executor.py
@ -45,7 +45,7 @@ class Executor:
                continue
            logits = model(feats)
            loss_type = args.get('criterion', 'max_pooling')
-            loss, acc = criterion(loss_type, logits, target, feats_lengths)
+            loss, acc = criterion(loss_type, logits, target, feats_lengths, min_duration)
            optimizer.zero_grad()
            loss.backward()
            grad_norm = clip_grad_norm_(model.parameters(), clip)