mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
2ca5d722b1
commit
83da8c41d7
Binary file not shown.
Binary file not shown.
@ -954,6 +954,8 @@ def train_one_epoch(
|
||||
)
|
||||
loss_info.reduce(loss.device)
|
||||
|
||||
logging.info(loss_info)
|
||||
|
||||
numel = params.world_size / (params.accum_grads * loss_info["utterances"])
|
||||
loss *= numel ## normalize loss over utts(batch size)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user