This commit is contained in:
Robert Sachunsky 2026-02-17 17:43:57 +01:00
parent 7bef8fa95a
commit c1b5cc92af

View file

@ -360,7 +360,7 @@ def run(_config,
if task == "enhancement": if task == "enhancement":
assert not is_loss_soft_dice, "for enhancement, soft_dice loss does not apply" assert not is_loss_soft_dice, "for enhancement, soft_dice loss does not apply"
assert not weighted_dice, "for enhancement, weighted loss does not apply" assert not weighted_loss, "for enhancement, weighted loss does not apply"
if continue_training: if continue_training:
custom_objects = dict() custom_objects = dict()
if is_loss_soft_dice: if is_loss_soft_dice: