2020-1-capstone-design2 / 2016104167

Go to a project
Toggle navigation Toggle navigation pinning
조현아
dad343e2 1 parent 46980b15

modified log dir

Inline Side-by-side
Showing 1 changed file with 1 additions and 1 deletions
...@@ -26,7 +26,7 @@ def train(**kwargs): ...@@ -26,7 +26,7 @@ def train(**kwargs):
26 26
27 print('\n[+] Create log dir') 27 print('\n[+] Create log dir')
28 model_name = get_model_name(args) 28 model_name = get_model_name(args)
29 - log_dir = os.path.join('/content/drive/My Drive/CD2 Project/runs', model_name) 29 + log_dir = os.path.join('logs', model_name)
30 os.makedirs(os.path.join(log_dir, 'model')) 30 os.makedirs(os.path.join(log_dir, 'model'))
31 json.dump(kwargs, open(os.path.join(log_dir, 'kwargs.json'), 'w')) 31 json.dump(kwargs, open(os.path.join(log_dir, 'kwargs.json'), 'w'))
32 writer = SummaryWriter(log_dir=log_dir) 32 writer = SummaryWriter(log_dir=log_dir)
......