from local

2025-12-11 06:55:27 +00:00 · 2023-02-25 15:51:57 +09:00 · 2023-02-25 15:51:57 +09:00 · ef9c392c5b
commit ef9c392c5b
parent 3d87d438fc
2 changed files with 5 additions and 2 deletions
--- a/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp
+++ b/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp
--- a/egs/librispeech/ASR/conformer_ctc2/transformer.py
+++ b/egs/librispeech/ASR/conformer_ctc2/transformer.py
@ -323,7 +323,10 @@ class Transformer(nn.Module):

        decoder_loss = self.decoder_criterion(pred_pad, ys_out_pad)
        
-        return decoder_loss
+        if return_output:
+            return pred_pad, decoder_loss
+        else:
+            return decoder_loss

    @torch.jit.export
    def decoder_nll(