From fcdfc4402f07ef87c8a65e8ae032a879821afa89 Mon Sep 17 00:00:00 2001 From: Juan Cobos Date: Wed, 5 Aug 2026 12:35:10 +0200 Subject: [PATCH] Add detector_eval_interval to train_from_coco to enable validation out of fixed eval_interval=10 for the detector as in base config --- deeplabcut/pose_estimation_pytorch/modelzoo/train_from_coco.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/deeplabcut/pose_estimation_pytorch/modelzoo/train_from_coco.py b/deeplabcut/pose_estimation_pytorch/modelzoo/train_from_coco.py index 2db3436752..01e89a03ce 100644 --- a/deeplabcut/pose_estimation_pytorch/modelzoo/train_from_coco.py +++ b/deeplabcut/pose_estimation_pytorch/modelzoo/train_from_coco.py @@ -37,6 +37,7 @@ def adaptation_train( batch_size: int = 8, detector_batch_size: int = 8, eval_interval: int | None = None, + detector_eval_interval: int | None = None, skip_detector: bool = False, ): setup_file_logging(Path(model_folder) / "log.txt") @@ -60,6 +61,8 @@ def adaptation_train( loader.model_cfg.detector.runner.snapshots.save_epochs = detector_save_epochs loader.model_cfg.detector.train_settings.batch_size = detector_batch_size loader.model_cfg.detector.train_settings.epochs = detector_epochs + if detector_eval_interval is not None: + loader.model_cfg.detector.runner.eval_interval = detector_eval_interval if eval_interval is not None: loader.model_cfg.runner.eval_interval = eval_interval