[Fix] Fix a makedirs bug (#2066)

PaddlePaddle · Apr 29, 2022 · 01ed9c3 · 01ed9c3
1 parent 1baf96e
commit 01ed9c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddleseg/core/train.py b/paddleseg/core/train.py
@@ -111,7 +111,7 @@ def train(model,
     if not os.path.isdir(save_dir):
         if os.path.exists(save_dir):
             os.remove(save_dir)
-        os.makedirs(save_dir)
+        os.makedirs(save_dir, exist_ok=True)
 
     # use amp
     if precision == 'fp16':