add comment regarding override_config

AllenCell · Jan 10, 2024 · 6346e23 · 6346e23
1 parent 2415cd2
commit 6346e23
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/allencell_ml_segmenter/services/prediction_service.py b/src/allencell_ml_segmenter/services/prediction_service.py
@@ -72,6 +72,7 @@ def _predict_model(self, _: Event) -> None:
         if continue_prediction:
             cyto_api: CytoDLModel = CytoDLModel()
             cyto_api.load_config_from_file(training_config)
+            # TODO: override config currently does not set train: False correctly. Retest once benji fixes the method.
             cyto_api.override_config(
                 self._build_default_prediction_overrides(
                     experiment_name, checkpoint_selected