add model name to config

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
+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