lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-10 00:59:34 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Clear evaluation callback after epoch ends

Browse Source
This commit is contained in:
mbsantiago 2025-09-01 08:56:38 +01:00
parent 71c2301c21
commit e0ecc3c3d1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/batdetect2/train/callbacks.py
View File

@ -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