Merge pull request #28 from wk910930/dev

Change os.mkdir() -> os.mkdirs()
AlexHex7 · Jul 10, 2017 · be9de01 · be9de01
2 parents 3418777 + 83c90f4
commit be9de01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -88,7 +88,7 @@ def log_print(text, color=None, on_color=None, attrs=None):
 optimizer = torch.optim.SGD(params[8:], lr=lr, momentum=momentum, weight_decay=weight_decay)
 
 if not os.path.exists(output_dir):
- os.mkdir(output_dir)
+ os.makedirs(output_dir)
 
 # tensorboad
 use_tensorboard = use_tensorboard and CrayonClient is not None