from local

This commit is contained in:
dohe0342 2022-12-26 14:54:37 +09:00
parent cf2483e69a
commit 200bdd4c8a
3 changed files with 1 additions and 3 deletions

Binary file not shown.

View File

@ -869,9 +869,7 @@ def compute_loss(
reduction="sum",
use_double_scores=params.use_double_scores,
)
#assert ctc_loss.requires_grad == is_training
ctc_loss.backward()
exit()
assert ctc_loss.requires_grad == is_training
loss += params.ctc_loss_scale * ctc_loss
info["ctc_loss"] = ctc_loss.detach().cpu().item()