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

2022-05-29 11:04:17 +08:00 · 2022-05-29 11:04:17 +08:00 · cff8a5fe26
commit cff8a5fe26
parent 41a3432198
1 changed files with 1 additions and 0 deletions
--- a/kws/utils/executor.py
+++ b/kws/utils/executor.py
@ -50,6 +50,7 @@ class Executor:
            grad_norm = clip_grad_norm_(model.parameters(), clip)
            if torch.isfinite(grad_norm):
                optimizer.step()
+            optimizer.zero_grad()
            if batch_idx % log_interval == 0:
                logging.debug(
                    'TRAIN Batch {}/{} loss {:.8f} acc {:.8f}'.format(