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):
print('\n[+] Create log dir')
model_name = get_model_name(args)
log_dir = os.path.join('/content/drive/My Drive/CD2 Project/runs', model_name)
log_dir = os.path.join('logs', 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)
......