From c0ea33473868d964bb1242207ac9011b6bde3d08 Mon Sep 17 00:00:00 2001 From: Tiance Wang Date: Fri, 24 Jun 2022 19:31:09 +0800 Subject: [PATCH] fix bug of concatenating list to tuple (#444) --- egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py index bf3917df0..0fa4b6907 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py @@ -1195,7 +1195,7 @@ class RandomCombine(nn.Module): ans = torch.matmul(stacked_inputs, weights) # ans: (*, num_channels) - ans = ans.reshape(inputs[0].shape[:-1] + [num_channels]) + ans = ans.reshape(inputs[0].shape[:-1] + (num_channels,)) # The following if causes errors for torch script in torch 1.6.0 # if __name__ == "__main__":