From 8fce9a05fc56b7efeee29898af5c7a0a34d332fc Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 17 May 2023 14:53:53 +0800 Subject: [PATCH] Revert batch size to 18 --- egs/libriheavy/LM/zipformer1/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/libriheavy/LM/zipformer1/train.py b/egs/libriheavy/LM/zipformer1/train.py index 173058096..a1f75ff2d 100755 --- a/egs/libriheavy/LM/zipformer1/train.py +++ b/egs/libriheavy/LM/zipformer1/train.py @@ -402,7 +402,7 @@ def get_params() -> AttributeDict: "warm_step": 2000, "env_info": get_env_info(), "bytes_per_segment": 2048, - "batch_size": 20, + "batch_size": 18, "train_file_list": "train.txt", "valid_file_list": "valid.txt", "num_workers": 4,