From 2f1d37772781beeebcf586f37a044420566397a3 Mon Sep 17 00:00:00 2001
From: Daniel Povey <dpovey@gmail.com>
Date: Thu, 4 May 2023 17:01:30 +0800
Subject: [PATCH] Reduce batch size so it fits in memory

---
 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 e6ddae4e8..1b9631ea5 100755
--- a/egs/libriheavy/LM/zipformer1/train.py
+++ b/egs/libriheavy/LM/zipformer1/train.py
@@ -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,