mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-01-10 17:19:34 +01:00
Fix logging issue
This commit is contained in:
parent
40f6b64611
commit
55f473c9ca
@ -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.")
|
||||||
|
|||||||
@ -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,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user