From b946c9a7abb070f3b8b022bb0ffb5a9e178c5d9f Mon Sep 17 00:00:00 2001 From: SaiYiKi <9478838+akira-l@users.noreply.github.com> Date: Tue, 23 Jul 2019 23:38:38 +0800 Subject: [PATCH] Update config.py --- config.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/config.py b/config.py index 87d9a6f..291c893 100644 --- a/config.py +++ b/config.py @@ -109,6 +109,8 @@ class LoadConfig(object): self.swap_num = args.swap_num self.save_dir = './net_model' + if not os.path.exists(self.save_dir): + os.mkdir(self.save_dir) self.backbone = args.backbone self.use_dcl = True