Modify example config to add name

This commit is contained in:
mbsantiago 2026-03-08 15:23:14 +00:00
parent f2d5088bec
commit 45e3cf1434
2 changed files with 2 additions and 2 deletions

View File

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

View File

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