from local

This commit is contained in:
dohe0342 2023-02-14 19:02:29 +09:00
parent 4958225ae9
commit 09a13083dd
2 changed files with 1 additions and 1 deletions

View File

@ -683,7 +683,7 @@ def compute_loss(
f"ctc_loss: {ctc_loss}\n"
f"att_loss: {att_loss}"
)
display_and_save_batch(batch, params=params, sp=graph_compiler.sp)
#display_and_save_batch(batch, params=params, sp=graph_compiler.sp)
ctc_loss = ctc_loss[ctc_loss_is_finite]
att_loss = att_loss[att_loss_is_finite]