From 3bd4695c160dded3e97a5b355c2e9daccd3c8d95 Mon Sep 17 00:00:00 2001 From: rohith Date: Mon, 12 Jul 2021 10:15:02 +0530 Subject: [PATCH] dir issue attempt-#1 --- main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 9e5cdd9..eddf511 100644 --- 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"),