from local

This commit is contained in:
dohe0342 2022-12-26 14:45:09 +09:00
parent e434e22672
commit bf33c6218b
3 changed files with 2 additions and 1 deletions

View File

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