diff --git a/egs/librispeech/ASR/.gtrans.sh.swp b/egs/librispeech/ASR/.gtrans.sh.swp index 60e90fd8b..f1f37764c 100644 Binary files a/egs/librispeech/ASR/.gtrans.sh.swp and b/egs/librispeech/ASR/.gtrans.sh.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp index f73ec2a9d..91aa1967f 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py index 6ea83883a..d4d5d38ee 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py @@ -201,11 +201,11 @@ class Conformer(EncoderInterface): ) # (T, N, C) x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C) - layer_output = [x.permute(1, 0, 2) for x in layer_output] + layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs] x = 0 for enum, alpha in enumerate(self.alpha): - x += self.sigmoid(alpha*layer_output[(enum+1)*self.group_layer_num-1]) + x += self.sigmoid(alpha*layer_outputs[(enum+1)*self.group_layer_num-1]) x = self.layer_norm(x/self.group_num)