From 45e3cf1434ecb107bc0b3b0642bcba1009020be0 Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Sun, 8 Mar 2026 15:23:14 +0000 Subject: [PATCH] Modify example config to add name --- example_data/config.yaml | 1 + src/batdetect2/models/backbones.py | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/example_data/config.yaml b/example_data/config.yaml index 7d908af..99d0a10 100644 --- a/example_data/config.yaml +++ b/example_data/config.yaml @@ -29,6 +29,7 @@ postprocess: top_k_per_sec: 200 model: + name: UNetBackbone input_height: 128 in_channels: 1 out_channels: 32 diff --git a/src/batdetect2/models/backbones.py b/src/batdetect2/models/backbones.py index 31feee0..07bc4b8 100644 --- a/src/batdetect2/models/backbones.py +++ b/src/batdetect2/models/backbones.py @@ -18,11 +18,10 @@ automatic padding to handle input sizes not perfectly divisible by the network's total downsampling factor. """ -from typing import Annotated, Literal, Tuple +from typing import Annotated, Literal, Tuple, Union import torch import torch.nn.functional as F -from example import Union from pydantic import Field from soundevent import data