Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add model name to experiemtn folder

This commit is contained in:
Eren Golge 2018-05-10 16:14:23 -07:00
parent 1770a9338c
commit 45bb9d9eec
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -42,7 +42,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'))

4
utils/generic_utils.py
View File

@ -34,11 +34,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