forked from nathangrinsztajn/DAG-scheduling
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathlog_utils.py
37 lines (31 loc) · 1.13 KB
/
log_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from config import config_enhanced
import os
from datetime import datetime
def name_env_dir(config_file):
name = ""
for k, v in config_file["env_settings"].items():
name += (k + "=" + str(v) + "_")
name += "seed=" + str(config_file['seed_env'])
return name
def name_mod_dir(config_file):
name = ""
for k, v in config_file["network_parameters"].items():
name += (k + "=" + str(v) + "_")
name = name[:-1]
return name
def name_dir(config_file):
date_srt = str(datetime.today())[:-7].replace(' ', '_')
env_str = name_env_dir(config_file)
mod_str = name_mod_dir(config_file)
return os.path.join(env_str, mod_str, date_srt)
def set_writer_dir(writer, config_file):
name_dir = name_env_dir(config_file)
current_dir = str(writer.get_logdir()).split('/')
new_dir_path = os.path.join(current_dir[0], name_dir, current_dir[1])
writer.log_dir = new_dir_path
# def name_model_sub_dir
if __name__ == '__main__':
print(name_env_dir(config_enhanced))
print(str(datetime.today())[:-7].replace(' ', '_'))
print(name_mod_dir(config_enhanced))
print(name_dir(config_enhanced))