deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-04 17:39:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug in tuning/utils.py: add optimizer.zero_grad() before loss.backward()

Browse Source
This commit is contained in:
Hongji Wang
2025-10-13 20:50:29 +08:00
parent 55ba6e2825
commit 64fb49e1c8
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tuning/utils.py
View File

@@ -240,6 +240,7 @@ def train(config,
loss = criterion(stacked, targets) loss = criterion(stacked, targets)
loss = (loss * masks).mean() loss = (loss * masks).mean()
optimizer.zero_grad()
loss.backward() loss.backward()
optimizer.step() optimizer.step()
losses.update(loss.item(), masks.numel()) losses.update(loss.item(), masks.numel())