fix race condition

mwalmsley · Nov 23, 2023 · 4d680d2 · 4d680d2
1 parent e60118b
commit 4d680d2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/zoobot/pytorch/training/train_with_pytorch_lightning.py b/zoobot/pytorch/training/train_with_pytorch_lightning.py
@@ -114,7 +114,10 @@ def train_default_zoobot_from_scratch(
 
     assert save_dir is not None
     if not os.path.isdir(save_dir):
-        os.mkdir(save_dir)
+        try:
+            os.mkdir(save_dir)
+        except FileExistsError():
+            pass # another gpu process may have just made it
     logging.info(f'Saving to {save_dir}')
 
     if color: