[TTS]VITS init sampler reverse, test=tts (#2843)

pull/2852/head
HuangLiangJie 3 years ago committed by GitHub
parent 478fd2593e
commit 140aed4b54
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -187,7 +187,7 @@ def main():
record["spk_emb"] = str(item["spk_emb"])
output_metadata.append(record)
output_metadata.sort(key=itemgetter('feats_lengths'))
output_metadata.sort(key=itemgetter('feats_lengths'), reverse=True)
output_metadata_path = Path(args.dumpdir) / "metadata.jsonl"
with jsonlines.open(output_metadata_path, 'w') as writer:
for item in output_metadata:

@ -166,7 +166,7 @@ def process_sentences(config,
if record:
results.append(record)
results.sort(key=itemgetter("feats_lengths"))
results.sort(key=itemgetter("feats_lengths"), reverse=True)
with jsonlines.open(output_dir / "metadata.jsonl", 'w') as writer:
for item in results:
writer.write(item)

@ -110,7 +110,7 @@ def train_sp(args, config):
train_sampler = ErnieSATSampler(
train_dataset,
batch_size=config.batch_size,
shuffle=True,
shuffle=False,
drop_last=True)
dev_sampler = ErnieSATSampler(
dev_dataset,

Loading…
Cancel
Save