mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change chunk-size setup

Browse Source
This commit is contained in:
Daniel Povey 2023-05-16 19:47:23 +08:00
parent a6eb45840a
commit 399a79ace6
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/libriheavy/LM/zipformer1/subformer.py
View File

@ -677,7 +677,7 @@ class SubformerEncoder(nn.Module):
assert seq_len % self.chunk_size == 0, (seq_len, self.chunk_size) assert seq_len % self.chunk_size == 0, (seq_len, self.chunk_size)
num_layers = len(self.layers) num_layers = len(self.layers)
chunk_indexes = [0, 1] * (num_layers + 1 // 2) chunk_indexes = [0, 1] * (num_layers + 1 // 2)
return [ self.chunk_size, self.chunk_size * 2 ], chunk_indexes[:num_layers] return [ self.chunk_size, seq_len ], chunk_indexes[:num_layers]
def _to_chunk_size(self, src: Tensor, chunk_size: int) -> Tensor: def _to_chunk_size(self, src: Tensor, chunk_size: int) -> Tensor:

2
egs/libriheavy/LM/zipformer1/train.py
View File

@ -395,7 +395,7 @@ def get_params() -> AttributeDict:
"warm_step": 2000, "warm_step": 2000,
"env_info": get_env_info(), "env_info": get_env_info(),
"bytes_per_segment": 2048, "bytes_per_segment": 2048,
"batch_size": 22, "batch_size": 20,
"train_file_list": "train.txt", "train_file_list": "train.txt",
"valid_file_list": "valid.txt", "valid_file_list": "valid.txt",
"num_workers": 4, "num_workers": 4,