alfabill commited on
Commit
bebb31e
·
verified ·
1 Parent(s): b159145

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +3 -3
run.py CHANGED
@@ -64,13 +64,13 @@ class Options():
64
  def updateOptions(self, phase):
65
 
66
  if phase == "correct_to_mask":
67
- self.checkpoints_dir = "checkpoints/cm.lib"
68
 
69
  elif phase == "maskref_to_maskdet":
70
- self.checkpoints_dir = "checkpoints/mm.lib"
71
 
72
  elif phase == "maskfin_to_nude":
73
- self.checkpoints_dir = "checkpoints/mn.lib"
74
 
75
  # process(cv_img, mode)
76
  # return:
 
64
  def updateOptions(self, phase):
65
 
66
  if phase == "correct_to_mask":
67
+ self.checkpoints_dir = "cm.lib"
68
 
69
  elif phase == "maskref_to_maskdet":
70
+ self.checkpoints_dir = "mm.lib"
71
 
72
  elif phase == "maskfin_to_nude":
73
+ self.checkpoints_dir = "mn.lib"
74
 
75
  # process(cv_img, mode)
76
  # return: