From dd3abebc4d1c3dc62674e60afaa36484d2b1d938 Mon Sep 17 00:00:00 2001 From: babysor00 Date: Sun, 27 Feb 2022 13:25:58 +0800 Subject: [PATCH] Fix bug of preparing fid --- ppg2mel/preprocess.py | 2 +- train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ppg2mel/preprocess.py b/ppg2mel/preprocess.py index 1009ffd..6da9054 100644 --- a/ppg2mel/preprocess.py +++ b/ppg2mel/preprocess.py @@ -100,7 +100,7 @@ def preprocess_dataset(datasets_root, dataset, out_dir, n_processes, ppg_encoder d_fid_file = out_dir.joinpath("dev_fidlist.txt").open("w", encoding="utf-8") e_fid_file = out_dir.joinpath("eval_fidlist.txt").open("w", encoding="utf-8") for file in sorted(out_dir.joinpath("f0").glob("*.npy")): - id = os.path.basename(file).rstrip(".f0.npy") + id = os.path.basename(file).split(".f0.npy")[0] if id.endswith("01"): d_fid_file.write(id + "\n") elif id.endswith("09"): diff --git a/train.py b/train.py index fed7501..5a6a06c 100644 --- a/train.py +++ b/train.py @@ -18,7 +18,7 @@ def main(): parser.add_argument('--name', default=None, type=str, help='Name for logging.') parser.add_argument('--logdir', default='log/', type=str, help='Logging path.', required=False) - parser.add_argument('--ckpdir', default='ckpt/', type=str, + parser.add_argument('--ckpdir', default='ppg2mel/saved_models/', type=str, help='Checkpoint path.', required=False) parser.add_argument('--outdir', default='result/', type=str, help='Decode output path.', required=False)