mirrors
/
DCL
mirror of https://github.com/JDAI-CV/DCL.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update config.py

master
SaiYiKi 2019-07-23 23:38:38 +08:00 committed by GitHub
parent 6203db874e
commit b946c9a7ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

@ -109,6 +109,8 @@ class LoadConfig(object):
self.swap_num = args.swap_num self.swap_num = args.swap_num
self.save_dir = './net_model' self.save_dir = './net_model'
if not os.path.exists(self.save_dir):
os.mkdir(self.save_dir)
self.backbone = args.backbone self.backbone = args.backbone
self.use_dcl = True self.use_dcl = True