2020-1-capstone-design2 / 2016104167

Go to a project
Toggle navigation Toggle navigation pinning
조현아
ee634d8d 1 parent 9c71da6c

mv log path

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