from local

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

View File

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