From fd3491ba1b5a0ccfac7667a0bfc9048d1de792cc Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Tue, 24 Aug 2021 02:56:41 +0000
Subject: [PATCH] fix dataloader batchsize and minibatchsize

---
 deepspeech/exps/u2_kaldi/model.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/deepspeech/exps/u2_kaldi/model.py b/deepspeech/exps/u2_kaldi/model.py
index 4f6ff4cb..46e5b4d9 100644
--- a/deepspeech/exps/u2_kaldi/model.py
+++ b/deepspeech/exps/u2_kaldi/model.py
@@ -228,7 +228,7 @@ class U2Trainer(Trainer):
             maxlen_in=float('inf'),
             maxlen_out=float('inf'),
             minibatches=0,
-            mini_batch_size=1,
+            mini_batch_size=self.args.nprocs,
             batch_count='auto',
             batch_bins=0,
             batch_frames_in=0,
@@ -247,7 +247,7 @@ class U2Trainer(Trainer):
             maxlen_in=float('inf'),
             maxlen_out=float('inf'),
             minibatches=0,
-            mini_batch_size=1,
+            mini_batch_size=self.args.nprocs,
             batch_count='auto',
             batch_bins=0,
             batch_frames_in=0,
@@ -263,7 +263,7 @@ class U2Trainer(Trainer):
             json_file=config.data.test_manifest,
             train_mode=False,
             sortagrad=False,
-            batch_size=config.collator.batch_size,
+            batch_size=config.decoding.batch_size,
             maxlen_in=float('inf'),
             maxlen_out=float('inf'),
             minibatches=0,
@@ -282,7 +282,7 @@ class U2Trainer(Trainer):
             json_file=config.data.test_manifest,
             train_mode=False,
             sortagrad=False,
-            batch_size=config.collator.batch_size,
+            batch_size=config.decoding.batch_size,
             maxlen_in=float('inf'),
             maxlen_out=float('inf'),
             minibatches=0,