reading predefined config

mlcommons · Jul 25, 2023 · 6ff7a9a · 6ff7a9a
1 parent 0b08214
commit 6ff7a9a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/testing/test_full.py b/testing/test_full.py
@@ -1234,6 +1234,9 @@ def test_train_losses_segmentation_rad_2d(device):
     parameters = populate_header_in_parameters(parameters, parameters["headers"])
     # loop through selected models and train for single epoch
     for loss_type in ["dc", "dc_log", "dcce", "dcce_logits", "tversky", "focal"]:
+        parameters = parseConfig(
+            testingDir + "/config_segmentation.yaml", version_check_flag=False
+        )
         parameters["loss_function"] = loss_type
         parameters["nested_training"]["testing"] = -5
         parameters["nested_training"]["validation"] = -5