wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

add model name to config

Browse Source
This commit is contained in:
Eren Golge
2018-04-27 04:48:21 -07:00
parent 3c701c9d10
commit f3bb9dd0f9
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config.json
+1
View File
@@ -1,4 +1,5 @@
{
"model_name": "guided-attn",
"num_mels": 80,
"num_freq": 1025,
"sample_rate": 22050,
train.py
+1 -1
View File
@@ -41,7 +41,7 @@ args = parser.parse_args()
c = load_config(args.config_path)
_ = os.path.dirname(os.path.realpath(__file__))
OUT_PATH = os.path.join(_, c.output_path)
OUT_PATH = create_experiment_folder(OUT_PATH)
OUT_PATH = create_experiment_folder(OUT_PATH, c.model_name)
CHECKPOINT_PATH = os.path.join(OUT_PATH, 'checkpoints')
shutil.copyfile(args.config_path, os.path.join(OUT_PATH, 'config.json'))
utils/generic_utils.py
+2 -2
View File
@@ -35,11 +35,11 @@ def get_commit_hash():
return commit
def create_experiment_folder(root_path):
def create_experiment_folder(root_path, model_name):
""" Create a folder with the current date and time """
date_str = datetime.datetime.now().strftime("%B-%d-%Y_%I:%M%p")
commit_hash = get_commit_hash()
output_folder = os.path.join(root_path, date_str + '-' + commit_hash)
output_folder = os.path.join(root_path, date_str + '-' + model_name + '-' + commit_hash)
os.makedirs(output_folder, exist_ok=True)
print(" > Experiment folder: {}".format(output_folder))
return output_folder