From b737d3d3944b553cbe75b7bb859071805ec88f1f Mon Sep 17 00:00:00 2001 From: michaelowenliu Date: Thu, 28 Apr 2022 20:13:40 +0800 Subject: [PATCH 1/2] fix a makedirs bug --- paddleseg/core/train.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/paddleseg/core/train.py b/paddleseg/core/train.py index a2c4072adc..ff2c6ddb62 100644 --- a/paddleseg/core/train.py +++ b/paddleseg/core/train.py @@ -109,9 +109,7 @@ def train(model, start_iter = resume(model, optimizer, resume_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': From 1c2046a71bd54804d8dbcb34488757e5b1fbeef6 Mon Sep 17 00:00:00 2001 From: cc <52520497+juncaipeng@users.noreply.github.com> Date: Fri, 29 Apr 2022 15:50:25 +0800 Subject: [PATCH 2/2] Update train.py --- paddleseg/core/train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddleseg/core/train.py b/paddleseg/core/train.py index ff2c6ddb62..fab5925944 100644 --- a/paddleseg/core/train.py +++ b/paddleseg/core/train.py @@ -109,6 +109,8 @@ def train(model, start_iter = resume(model, optimizer, resume_model) if not os.path.isdir(save_dir): + if os.path.exists(save_dir): + os.remove(save_dir) os.makedirs(save_dir, exist_ok=True) # use amp