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

Reduce batch size so it fits in memory

Browse Source
This commit is contained in:
Daniel Povey 2023-05-04 17:01:30 +08:00
parent f0264bed1b
commit 2f1d377727
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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