mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove unused codes. (#821)

Browse Source
This commit is contained in:
kobenaxie 2023-01-06 11:16:22 +08:00 committed by GitHub
parent b9626f2e06
commit 9a9c5a0f9b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer2.py
View File

@ -1512,24 +1512,6 @@ class EmformerEncoder(nn.Module):
)
return states
attn_caches = [
[
torch.zeros(self.memory_size, self.d_model, device=device),
torch.zeros(self.left_context_length, self.d_model, device=device),
torch.zeros(self.left_context_length, self.d_model, device=device),
]
for _ in range(self.num_encoder_layers)
]
conv_caches = [
torch.zeros(self.d_model, self.cnn_module_kernel - 1, device=device)
for _ in range(self.num_encoder_layers)
]
states: Tuple[List[List[torch.Tensor]], List[torch.Tensor]] = (
attn_caches,
conv_caches,
)
return states
class Emformer(EncoderInterface):
def __init__(