Skip to content

Commit 82a2f8b

Browse files
authored
fix finetune batch size (#2457) (#2460)
* fix batch_size, test=tts
1 parent 615da21 commit 82a2f8b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/other/tts_finetune/tts3/local/finetune.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,10 +131,10 @@ def train_sp(args, config):
131131
converters=converters, )
132132

133133
# collate function and dataloader
134-
134+
train_batch_size = min(len(train_metadata), config.batch_size)
135135
train_sampler = DistributedBatchSampler(
136136
train_dataset,
137-
batch_size=config.batch_size,
137+
batch_size=train_batch_size,
138138
shuffle=True,
139139
drop_last=True)
140140

0 commit comments

Comments
 (0)