From 8031ade3b55b9188e8520d8c04ac64ba84e153c4 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 2 Feb 2023 14:09:13 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc/.transformer.py.swp | Bin 69632 -> 69632 bytes egs/aishell/ASR/conformer_ctc/transformer.py | 5 +++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp index 34ffe002d220d4d0ff57bfdf39ed496a741c787c..a0a71f3c9d8e9e6cc2a581254cb05b4802b65b19 100644 GIT binary patch delta 215 zcmZozz|ydQMJ&l6%+puFQqO<^2m}}y^jvQzFJav%W+lKVwmDE>vpz2~F9Sn45QAjJ zCJQ>ai%Rh^Fnj@uoCe|rK->t#<(qRI1bHX__Trzs(s!zGK~ZL2iAHK(a(+r`QG8y0 zQLg4@Q@`^QS?yRE7>p)!K3H!ymz{y37Kp`x_#GPq!&@Le55)6;crFlE191ot2LrJQ l5E}!r5!+@)7B|+(4$qFUSjB?V-s1uSQ!}XCUZVm zZ?>47fuR|QWr6q?8w0~nAifR6tAKbV5Vr$y0uaXou>%m>1F;?3W=57E*6Deyj2x3~ Pp6#3TeCg)Z&qFl;Ch0Am diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py index 982abc6c0..90aca750d 100644 --- a/egs/aishell/ASR/conformer_ctc/transformer.py +++ b/egs/aishell/ASR/conformer_ctc/transformer.py @@ -113,6 +113,7 @@ class Transformer(nn.Module): # num_layers=num_encoder_layers, # norm=encoder_norm, #) + print(encoder_norm) self.encoder = TransfEncoder( encoder_layer=encoder_layer, num_layers=num_encoder_layers, @@ -491,10 +492,10 @@ class TransfEncoder(nn.TransformerEncoder): output = torch._nested_tensor_from_mask(output, src_key_padding_mask.logical_not(), mask_check=False) src_key_padding_mask_for_layers = None - outputs = [] + layer_outputs = [] for mod in self.layers: output = mod(output, src_mask=mask, src_key_padding_mask=src_key_padding_mask_for_layers) - outputs.append(output) + layer_outputs.append(output) if convert_to_nested: output = output.to_padded_tensor(0.)