Skip to content

Commit a606e2f

Browse files
committed
Update checkpoint variables togeteher with MODEL.LOAD_CHECKPOINT.INPUT when reusing a configuration
1 parent 2be5b41 commit a606e2f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ui_utils.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2588,7 +2588,7 @@ def create_yaml_file(main_window):
25882588
biapy_config['MODEL']['LOAD_CHECKPOINT'] = True
25892589
biapy_config['MODEL']["SOURCE"] = "biapy"
25902590
biapy_config['PATHS'] = {}
2591-
if get_text(main_window.ui.PATHS__CHECKPOINT_FILE__INPUT) != "":
2591+
if get_text(main_window.ui.PATHS__CHECKPOINT_FILE__INPUT) != "" and get_text(main_window.ui.PATHS__CHECKPOINT_FILE__INPUT) != "model_weights.pth":
25922592
biapy_config['PATHS']['CHECKPOINT_FILE'] = get_text(main_window.ui.PATHS__CHECKPOINT_FILE__INPUT, strip=False)
25932593
if int(get_text(main_window.ui.MODEL__SAVE_CKPT_FREQ__INPUT)) != -1:
25942594
biapy_config['MODEL']['SAVE_CKPT_FREQ'] = int(get_text(main_window.ui.MODEL__SAVE_CKPT_FREQ__INPUT))
@@ -3459,6 +3459,10 @@ def analyze_dict(self, conf, sep=""):
34593459
v = "I want to check other online sources"
34603460
other_widgets_to_set.append("LOAD_PRETRAINED_MODEL__INPUT")
34613461
other_widgets_values_to_set.append("Yes")
3462+
elif widget_name == "MODEL__LOAD_CHECKPOINT__INPUT":
3463+
if self.biapy_pretrained:
3464+
other_widgets_to_set.append("LOAD_PRETRAINED_MODEL__INPUT")
3465+
other_widgets_values_to_set.append("Yes")
34623466
elif widget_name == "MODEL__BMZ__SOURCE_MODEL_ID__INPUT":
34633467
if self.bmz_pretrained:
34643468
v += " (BioImage Model Zoo)"

0 commit comments

Comments
 (0)