fix test_full dynunet model error "kernel_size"

mlcommons · Jun 1, 2024 · 86e4188 · 86e4188
1 parent dec2b3b
commit 86e4188
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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