From 51f0fe6dc364dc8faa68577b236c6813e697382f Mon Sep 17 00:00:00 2001 From: "Wall.E" <1424127847@qq.com> Date: Wed, 13 Jul 2022 23:50:48 +0800 Subject: [PATCH] fixed the parameter transfer problem for criterion (#75) * fixed the parameter transfer problem for criterion Co-authored-by: yangyyt --- kws/utils/executor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/kws/utils/executor.py b/kws/utils/executor.py index d30fb0a..822ed14 100644 --- a/kws/utils/executor.py +++ b/kws/utils/executor.py @@ -45,7 +45,8 @@ 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)