from local

2023-01-08 18:46:04 +09:00 · 2023-01-08 18:46:04 +09:00 · f84a12540a
commit f84a12540a
parent 1b607a9c0f
2 changed files with 1 additions and 1 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
@ -689,7 +689,7 @@ class ConformerEncoder(nn.Module):
            )
            layer_outputs.append(output)

-        return output
+        return output, layer_outputs

    @torch.jit.export
    def chunk_forward(