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

fix bug in kws/utils/executor.py (#72)

Browse Source
This commit is contained in:
chmod740 2022-05-29 11:04:17 +08:00
parent 41a3432198
commit cff8a5fe26
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
kws/utils/executor.py
View File

@ -50,6 +50,7 @@ class Executor:
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):
optimizer.step() optimizer.step()
optimizer.zero_grad()
if batch_idx % log_interval == 0: if batch_idx % log_interval == 0:
logging.debug( logging.debug(
'TRAIN Batch {}/{} loss {:.8f} acc {:.8f}'.format( 'TRAIN Batch {}/{} loss {:.8f} acc {:.8f}'.format(