AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update executor.py

Browse Source
This commit is contained in:
xiaohou 2021-11-18 15:52:16 +08:00 committed by GitHub
parent 7ba75f6fe1
commit 24a0de00b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kws/utils/executor.py
View File

@ -51,8 +51,7 @@ class Executor:
if num_utts == 0: if num_utts == 0:
continue continue
logits = model(feats) logits = model(feats)
loss, acc = criterion(logits, target, feats_lengths, loss, acc = criterion(logits, target, feats_lengths, min_duration)
min_duration)
loss.backward() loss.backward()
grad_norm = clip_grad_norm_(model.parameters(), clip) grad_norm = clip_grad_norm_(model.parameters(), clip)
if torch.isfinite(grad_norm): if torch.isfinite(grad_norm):