From d113b74bb99f1fab33d0279306c066bf6a5c6c89 Mon Sep 17 00:00:00 2001 From: WongLaw Date: Wed, 19 Oct 2022 09:24:18 +0000 Subject: [PATCH] Revised Rhythm Prediction Function 6, test=tts --- examples/other/rhy/README.md | 4 +++- examples/other/rhy/local/pre_for_sp_aishell.py | 3 ++- examples/other/rhy/local/pre_for_sp_csmsc.py | 3 ++- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/examples/other/rhy/README.md b/examples/other/rhy/README.md index 58a6c3e5f..8fdffc914 100644 --- a/examples/other/rhy/README.md +++ b/examples/other/rhy/README.md @@ -18,6 +18,8 @@ ./run.sh --stage 3 --stop-stage 3 ``` ## Pretrained Model -The pretrained model can be downloaded below, and you should put it into `exp/YOUREXP/checkpoints` folder: +The pretrained model can be downloaded here: [snapshot_iter_2600.pdz](https://paddlespeech.bj.bcebos.com/Rhy_Prediction/snapshot_iter_2600.pdz) + +And you should put it into `exp/YOUREXP/checkpoints` folder. diff --git a/examples/other/rhy/local/pre_for_sp_aishell.py b/examples/other/rhy/local/pre_for_sp_aishell.py index 360efc1d5..a2a716683 100644 --- a/examples/other/rhy/local/pre_for_sp_aishell.py +++ b/examples/other/rhy/local/pre_for_sp_aishell.py @@ -26,7 +26,8 @@ def pre_and_write(data, file): def main(): - parser = argparse.ArgumentParser(description="Train a Rhy prediction model.") + parser = argparse.ArgumentParser( + description="Train a Rhy prediction model.") parser.add_argument("--data", type=str, default="label_train-set.txt") parser.add_argument( "--processed_path", type=str, default="../data/rhy_predict") diff --git a/examples/other/rhy/local/pre_for_sp_csmsc.py b/examples/other/rhy/local/pre_for_sp_csmsc.py index 5ada47114..0a96092c1 100644 --- a/examples/other/rhy/local/pre_for_sp_csmsc.py +++ b/examples/other/rhy/local/pre_for_sp_csmsc.py @@ -24,7 +24,8 @@ def pre_and_write(data, file): def main(): - parser = argparse.ArgumentParser(description="Train a Rhy prediction model.") + parser = argparse.ArgumentParser( + description="Train a Rhy prediction model.") parser.add_argument("--data", type=str, default="label_train-set.txt") parser.add_argument( "--processed_path", type=str, default="../data/rhy_predict")