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)