未验证 提交 140aed4b 编写于 作者: H HuangLiangJie 提交者: GitHub

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

上级 478fd259
...@@ -187,7 +187,7 @@ def main(): ...@@ -187,7 +187,7 @@ def main():
record["spk_emb"] = str(item["spk_emb"]) record["spk_emb"] = str(item["spk_emb"])
output_metadata.append(record) 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" output_metadata_path = Path(args.dumpdir) / "metadata.jsonl"
with jsonlines.open(output_metadata_path, 'w') as writer: with jsonlines.open(output_metadata_path, 'w') as writer:
for item in output_metadata: for item in output_metadata:
......
...@@ -166,7 +166,7 @@ def process_sentences(config, ...@@ -166,7 +166,7 @@ def process_sentences(config,
if record: if record:
results.append(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: with jsonlines.open(output_dir / "metadata.jsonl", 'w') as writer:
for item in results: for item in results:
writer.write(item) writer.write(item)
......
...@@ -110,7 +110,7 @@ def train_sp(args, config): ...@@ -110,7 +110,7 @@ def train_sp(args, config):
train_sampler = ErnieSATSampler( train_sampler = ErnieSATSampler(
train_dataset, train_dataset,
batch_size=config.batch_size, batch_size=config.batch_size,
shuffle=True, shuffle=False,
drop_last=True) drop_last=True)
dev_sampler = ErnieSATSampler( dev_sampler = ErnieSATSampler(
dev_dataset, dev_dataset,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册