From 86e41888047c348dfd01b86d5f408a99118813d0 Mon Sep 17 00:00:00 2001 From: "V.Malefioudakis" Date: Sat, 1 Jun 2024 15:25:56 +0300 Subject: [PATCH] fix test_full dynunet model error "kernel_size" --- testing/test_full.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/testing/test_full.py b/testing/test_full.py index b8ce0aca2..10eaaffab 100644 --- a/testing/test_full.py +++ b/testing/test_full.py @@ -372,6 +372,12 @@ def test_train_segmentation_rad_3d(device): parameters["model"]["converter_type"] = random.choice( ["acs", "soft", "conv3d"] ) + + if model == "dynunet": + # More info: https://github.com/Project-MONAI/MONAI/blob/96bfda00c6bd290297f5e3514ea227c6be4d08b4/tests/test_dynunet.py + parameters["model"]["kernel_size"] = (3, 3, 3, 1) + parameters["model"]["strides"] = (1, 1, 1, 1) + parameters["model"]["deep_supervision"] = False parameters["model"]["architecture"] = model parameters["nested_training"]["testing"] = -5 parameters["nested_training"]["validation"] = -5