From e0ecc3c3d1c85dd9c94b3dc03546ba15c409171f Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Mon, 1 Sep 2025 08:56:38 +0100 Subject: [PATCH] Clear evaluation callback after epoch ends --- src/batdetect2/train/callbacks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/batdetect2/train/callbacks.py b/src/batdetect2/train/callbacks.py index 342d88a..d5bdf31 100644 --- a/src/batdetect2/train/callbacks.py +++ b/src/batdetect2/train/callbacks.py @@ -108,7 +108,8 @@ class ValidationMetrics(Callback): trainer: Trainer, pl_module: LightningModule, ) -> None: - self._matches = [] + self._clip_annotations = [] + self._predictions = [] return super().on_validation_epoch_start(trainer, pl_module) def on_validation_batch_end( # type: ignore