From 7ef00988e54e3a637402cebfe58a8eeb23bd3569 Mon Sep 17 00:00:00 2001
From: liangym <34430015+lym0302@users.noreply.github.com>
Date: Mon, 26 Sep 2022 19:09:49 +0800
Subject: [PATCH] fix finetune batch size (#2457)

* fix batch_size, test=tts
---
 examples/other/tts_finetune/tts3/local/finetune.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/other/tts_finetune/tts3/local/finetune.py b/examples/other/tts_finetune/tts3/local/finetune.py
index 496c2355b..814497aaa 100644
--- a/examples/other/tts_finetune/tts3/local/finetune.py
+++ b/examples/other/tts_finetune/tts3/local/finetune.py
@@ -131,10 +131,10 @@ def train_sp(args, config):
         converters=converters, )
 
     # collate function and dataloader
-
+    train_batch_size = min(len(train_metadata), config.batch_size)
     train_sampler = DistributedBatchSampler(
         train_dataset,
-        batch_size=config.batch_size,
+        batch_size=train_batch_size,
         shuffle=True,
         drop_last=True)