mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-04-11 16:12:25 +09:00
parent 875c436bf3
commit 1754bf0cd7
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
View File

Binary file not shown.

2
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
View File

@ -1063,7 +1063,7 @@ def train_one_epoch(
try: loss_info.reduce(loss.device) try: loss_info.reduce(loss.device)
except: pass except: pass
logging.info(loss_info) print(loss_info)
numel = params.world_size / (params.accum_grads * loss_info["utterances"]) numel = params.world_size / (params.accum_grads * loss_info["utterances"])
loss *= numel ## normalize loss over utts(batch size) loss *= numel ## normalize loss over utts(batch size)