From 9003703d6ce4143a4900b4ce9854ce0ea9f558a0 Mon Sep 17 00:00:00 2001 From: yangyyt Date: Wed, 13 Jul 2022 21:42:52 +0800 Subject: [PATCH] fixed the parameter transfer problem for criterion --- kws/utils/executor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kws/utils/executor.py b/kws/utils/executor.py index d30fb0a..4091590 100644 --- 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)