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 14:15:44 +09:00
parent de2a41d86c
commit 95c3d5b854
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

21
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
View File

@ -913,12 +913,21 @@ def compute_loss(
if decode:
model.eval()
with torch.no_grad():
hypos = model.module.decode(
x=feature,
x_lens=feature_lens,
y=y,
sp=sp
)
try:
hypos = model.module.decode(
x=feature,
x_lens=feature_lens,
y=y,
sp=sp
)
except:
hypos = model.decode(
x=feature,
x_lens=feature_lens,
y=y,
sp=sp
)
logging.info(f'ref: {batch["supervisions"]["text"][0]}')
logging.info(f'hyp: {" ".join(hypos[0])}')
model.train()