dir issue attempt-yuzhoucw#1

Rohith-2 · Jul 12, 2021 · 3bd4695 · 3bd4695
1 parent 443db04
commit 3bd4695
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -96,7 +96,11 @@
     if args.mode == "test":
         out_dir = os.path.dirname(args.pretrain_path)
         out_dir_img = os.path.join(out_dir, "images", "test")
-        #os.mkdir(out_dir_img)
+
+        try:
+            os.mkdir(out_dir_img)
+        except:
+            continue
 
         # load data
         test_loader = dataloader.get_dataloader(os.path.join(args.data_dir, "testA"),