From f6cf18e747bfa084e1f50ea3d2cfebaf84ca937c Mon Sep 17 00:00:00 2001 From: liangym Date: Mon, 26 Sep 2022 10:56:52 +0000 Subject: [PATCH] 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)