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