from local

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

View File

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