from local

This commit is contained in:
dohe0342 2022-12-10 11:06:48 +09:00
parent d29ee9df25
commit 09d287c943
2 changed files with 1 additions and 1 deletions

View File

@ -762,7 +762,7 @@ def compute_loss(
info["loss"] = loss.detach().cpu().item()
info["simple_loss"] = simple_loss.detach().cpu().item()
info["pruned_loss"] = pruned_loss.detach().cpu().item()
info["ctc_loss"] = ctc_loss.detach().cpu().item()
#info["ctc_loss"] = ctc_loss.detach().cpu().item()
return loss, info