Fix logging issue

This commit is contained in:
mbsantiago 2025-08-31 19:06:09 +01:00
parent 40f6b64611
commit 55f473c9ca
2 changed files with 1 additions and 2 deletions

View File

@ -70,7 +70,7 @@ def train_command(
val_annotations = load_dataset_from_config(val_dataset) val_annotations = load_dataset_from_config(val_dataset)
logger.debug( logger.debug(
"Loaded {num_annotations} validation examples", "Loaded {num_annotations} validation examples",
num_files=len(val_annotations), num_annotations=len(val_annotations),
) )
else: else:
logger.debug("No validation directory provided.") logger.debug("No validation directory provided.")

View File

@ -22,7 +22,6 @@ from torch import nn
from batdetect2.configs import BaseConfig from batdetect2.configs import BaseConfig
from batdetect2.models.blocks import ( from batdetect2.models.blocks import (
LayerConfig,
SelfAttentionConfig, SelfAttentionConfig,
VerticalConv, VerticalConv,
build_layer_from_config, build_layer_from_config,