diff --git a/train_concap.py b/train_concap.py
index d7cc70b..a0ac3e4 100644
--- a/train_concap.py
+++ b/train_concap.py
@@ -272,8 +272,7 @@ def main():
         default_gpu = True
 
     if default_gpu:
-        if not os.path.exists(savePath):
-            os.makedirs(savePath)
+        os.makedirs(savePath, exist_ok=True)
 
     config = BertConfig.from_json_file(args.config_file)