generic train.py for multiple architectures set on config.json

This commit is contained in:
Eren Golge
2019-03-06 13:11:22 +01:00
parent a4474abd83
commit 08162157ee
2 changed files with 115 additions and 125 deletions
+4 -4
View File
@@ -48,10 +48,10 @@ def get_commit_hash():
def create_experiment_folder(root_path, model_name, debug):
""" Create a folder with the current date and time """
date_str = datetime.datetime.now().strftime("%B-%d-%Y_%I+%M%p")
if debug:
commit_hash = 'debug'
else:
commit_hash = get_commit_hash()
# if debug:
# commit_hash = 'debug'
# else:
commit_hash = get_commit_hash()
output_folder = os.path.join(
root_path, model_name + '-' + date_str + '-' + commit_hash)
os.makedirs(output_folder, exist_ok=True)