fix batch_size, test=tts

pull/2457/head
liangym 3 years ago
parent 697e1f70a5
commit f6cf18e747

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