Update config.py

2019-07-23 23:38:38 +08:00 · 2019-07-23 23:38:38 +08:00 · b946c9a7ab
parent 6203db874e
commit b946c9a7ab
1 changed files with 2 additions and 0 deletions
--- 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