fix testscript invalid config: only add detector updates for td models

deruyter92 · deruyter92 · commit 139dd99ea7cf · 2026-02-18T08:30:59.000+01:00
diff --git a/examples/testscript_pytorch_multi_animal.py b/examples/testscript_pytorch_multi_animal.py
@@ -67,17 +67,25 @@ def main(
                     "runner.device": device,
                     "runner.snapshots.save_epochs": save_epochs,
                     "runner.snapshots.max_snapshots": max_snapshots_to_keep,
-                    "detector.train_settings.display_iters": 1,
-                    "detector.train_settings.epochs": detector_epochs,
-                    "detector.train_settings.batch_size": detector_batch_size,
-                    "detector.train_settings.dataloader_workers": 0,
-                    "detector.runner.snapshots.save_epochs": save_epochs,
-                    "detector.runner.snapshots.max_snapshots": max_snapshots_to_keep,
                     "logger": logger,
                 }
+
+                # Only add detector config updates for top-down models
+                if is_model_top_down(net_type):
+                    pytorch_cfg_updates.update({
+                        "detector.train_settings.display_iters": 1,
+                        "detector.train_settings.epochs": detector_epochs,
+                        "detector.train_settings.batch_size": detector_batch_size,
+                        "detector.train_settings.dataloader_workers": 0,
+                        "detector.runner.snapshots.save_epochs": save_epochs,
+                        "detector.runner.snapshots.max_snapshots": max_snapshots_to_keep,
+                    })
+                
+                # Only add conditional top-down config updates for conditional top-down models
                 if is_model_cond_top_down(net_type):
                     pytorch_cfg_updates["inference.conditions.shuffle"] = conditions_shuffle
                     pytorch_cfg_updates["inference.conditions.snapshot_index"] = -1
+
                 run(
                     config_path=config_path,
                     train_fraction=train_frac,
