fix finetune batch size (#2457)

* fix batch_size, test=tts
pull/2466/head
liangym 2 years ago committed by GitHub
parent e65622548d
commit 63f70c9fd0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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)

Loading…
Cancel
Save