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):
print('\n[+] Create log dir')
model_name = get_model_name(args)
log_dir = os.path.join('./runs', model_name)
log_dir = os.path.join('/content/drive/My Drive/CD2 Project/runs', model_name)
os.makedirs(os.path.join(log_dir, 'model'))
json.dump(kwargs, open(os.path.join(log_dir, 'kwargs.json'), 'w'))
writer = SummaryWriter(log_dir=log_dir)
......