from local

This commit is contained in:
dohe0342 2023-02-15 11:32:45 +09:00
parent 90710829a2
commit 4451575bb5
6 changed files with 1 additions and 1 deletions

View File

@ -738,7 +738,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]