mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-04-04 15:20:19 +02:00
Modify example config to add name
This commit is contained in:
parent
f2d5088bec
commit
45e3cf1434
@ -29,6 +29,7 @@ postprocess:
|
|||||||
top_k_per_sec: 200
|
top_k_per_sec: 200
|
||||||
|
|
||||||
model:
|
model:
|
||||||
|
name: UNetBackbone
|
||||||
input_height: 128
|
input_height: 128
|
||||||
in_channels: 1
|
in_channels: 1
|
||||||
out_channels: 32
|
out_channels: 32
|
||||||
|
|||||||
@ -18,11 +18,10 @@ automatic padding to handle input sizes not perfectly divisible by the
|
|||||||
network's total downsampling factor.
|
network's total downsampling factor.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from typing import Annotated, Literal, Tuple
|
from typing import Annotated, Literal, Tuple, Union
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
import torch.nn.functional as F
|
import torch.nn.functional as F
|
||||||
from example import Union
|
|
||||||
from pydantic import Field
|
from pydantic import Field
|
||||||
from soundevent import data
|
from soundevent import data
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user