From de88ade6d6185481233ef9d18edeef606901fa21 Mon Sep 17 00:00:00 2001 From: Phil Chlap Date: Wed, 13 Dec 2023 11:41:55 +1100 Subject: [PATCH] Allow setting fold in nnUNet service --- services/nnunet/service.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/services/nnunet/service.py b/services/nnunet/service.py index de1ef4f8..af101cff 100644 --- a/services/nnunet/service.py +++ b/services/nnunet/service.py @@ -28,6 +28,7 @@ "task": "TaskXXX", "config": "2d", "trainer": None, + "fold": None, "clean_sup_slices": False, } @@ -98,7 +99,10 @@ def nnunet_service(data_objects, working_dir, settings): settings["config"], ] - if settings["trainer"]: + if "fold" in settings and settings["fold"]: + command += ["-f", settings["fold"]] + + if "trainer" in settings and settings["trainer"]: command += ["-tr", settings["trainer"]] logger.info("Running command: %s", command)