Fix bug of preparing fid

This commit is contained in:
babysor00 2022-02-27 13:25:58 +08:00
parent eeee32f3e3
commit dd3abebc4d
2 changed files with 2 additions and 2 deletions

View File

@ -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"):

View File

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