fix finetune batch size (#2457) (#2460)

* fix batch_size, test=tts
2 years ago · 82a2f8b70e
parent 615da21102
commit 82a2f8b70e
1 changed files with 2 additions and 2 deletions
--- 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)