StarMirror/MockingBird
1
0
Fork 0
mirror of https://github.com/babysor/MockingBird.git synced 2024-03-22 13:11:31 +08:00
Code Issues Projects Releases Wiki Activity

Fix bug of preparing fid

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppg2mel/preprocess.py
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") 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") e_fid_file = out_dir.joinpath("eval_fidlist.txt").open("w", encoding="utf-8")
for file in sorted(out_dir.joinpath("f0").glob("*.npy")): 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"): if id.endswith("01"):
d_fid_file.write(id + "\n") d_fid_file.write(id + "\n")
elif id.endswith("09"): elif id.endswith("09"):

2
train.py
View File

@ -18,7 +18,7 @@ def main():
parser.add_argument('--name', default=None, type=str, help='Name for logging.') parser.add_argument('--name', default=None, type=str, help='Name for logging.')
parser.add_argument('--logdir', default='log/', type=str, parser.add_argument('--logdir', default='log/', type=str,
help='Logging path.', required=False) 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) help='Checkpoint path.', required=False)
parser.add_argument('--outdir', default='result/', type=str, parser.add_argument('--outdir', default='result/', type=str,
help='Decode output path.', required=False) help='Decode output path.', required=False)