hysts commited on
Commit
385c783
1 Parent(s): 6be65e1
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -65,6 +65,7 @@ class Model:
65
  self.task_name = ''
66
 
67
  self.model_dir = pathlib.Path(model_dir)
 
68
 
69
  self.use_lightweight = use_lightweight
70
  if use_lightweight:
 
65
  self.task_name = ''
66
 
67
  self.model_dir = pathlib.Path(model_dir)
68
+ self.model_dir.mkdir(exist_ok=True, parents=True)
69
 
70
  self.use_lightweight = use_lightweight
71
  if use_lightweight: