diff --git a/config/template.py b/config/template.py index 03cd9bb..d6661ac 100755 --- a/config/template.py +++ b/config/template.py @@ -33,6 +33,7 @@ class Config: #### PATH #### ROOT_DIR = os.path.dirname(os.path.abspath("__file__")) DATA_DIR = ROOT_DIR + "/" + # Change datalist.jsonl to datalist_mfa.jsonl if you have run MFA train_data_path = DATA_DIR + "/train/datalist.jsonl" valid_data_path = DATA_DIR + "/valid/datalist.jsonl" output_directory = ROOT_DIR + "/" @@ -44,7 +45,7 @@ class Config: bert_path = 'WangZeJun/simbert-base-chinese' token_list_path = ROOT_DIR + "//tokenlist" style_encoder_ckpt = ROOT_DIR + "/outputs/style_encoder/ckpt/checkpoint_163431" - tmp_dir = ROOT_DIR + "/tmp" + tmp_dir = output_directory + "/tmp" model_config_path = ROOT_DIR + "/config/joint/config.yaml" #### Model #### diff --git a/data/LJspeech/README.md b/data/LJspeech/README.md index c1d2ff7..7c5c200 100644 --- a/data/LJspeech/README.md +++ b/data/LJspeech/README.md @@ -170,7 +170,7 @@ Training tips: tensorboard --logdir=exp/LJspeech ``` - The model checkpoints are saved at `exp/LJspeech/ckpt`. -- The bert features are extracted in the first epoch and saved in `tmp/` folder, you can change the path in `exp/LJspeech/config/config.py`. +- The bert features are extracted in the first epoch and saved in `exp/LJspeech/tmp/` folder, you can change the path in `exp/LJspeech/config/config.py`. ### Step5 Inference