lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 14:41:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update model config in example config

Browse Source
This commit is contained in:
mbsantiago 2025-06-26 10:04:00 -06:00
parent 363eb9fb2f
commit 647468123e
1 changed files with 35 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
config.yaml
View File

@ -79,12 +79,36 @@ labels:
sigma: 3 sigma: 3
model: model:
name: Net2DFast
input_height: 128 input_height: 128
encoder_channels: [1, 32, 64, 128] in_channels: 1
bottleneck_channels: 256
decoder_channels: [256, 64, 32, 32]
out_channels: 32 out_channels: 32
encoder:
layers:
- block_type: FreqCoordConvDown
out_channels: 32
- block_type: FreqCoordConvDown
out_channels: 64
- block_type: LayerGroup
layers:
- block_type: FreqCoordConvDown
out_channels: 128
- block_type: ConvBlock
out_channels: 256
bottleneck:
channels: 256
self_attention: true
decoder:
layers:
- block_type: FreqCoordConvUp
out_channels: 64
- block_type: FreqCoordConvUp
out_channels: 32
- block_type: LayerGroup
layers:
- block_type: FreqCoordConvUp
out_channels: 32
- block_type: ConvBlock
out_channels: 32
train: train:
batch_size: 8 batch_size: 8
@ -104,29 +128,31 @@ train:
optimizer: optimizer:
learning_rate: 0.001 learning_rate: 0.001
t_max: 100 t_max: 100
logger:
logger_type: dvclive
augmentations: augmentations:
steps: steps:
- augmentation_type: mix - augmentation_type: mix_audio
probability: 0.2 probability: 0.2
min_weight: 0.3 min_weight: 0.3
max_weight: 0.7 max_weight: 0.7
- agumentation_type: echo - augmentation_type: add_echo
probability: 0.2 probability: 0.2
max_delay: 0.005 max_delay: 0.005
min_weight: 0.0 min_weight: 0.0
max_weight: 1.0 max_weight: 1.0
- augmentation_type: volume - augmentation_type: scale_volume
probability: 0.2 probability: 0.2
min_scaling: 0.0 min_scaling: 0.0
max_scaling: 2.0 max_scaling: 2.0
- augmentation_type: warp - augmentation_type: warp
probability: 0.2 probability: 0.2
delta: 0.04 delta: 0.04
- augmentation_type: time_mask - augmentation_type: mask_time
probability: 0.2 probability: 0.2
max_perc: 0.05 max_perc: 0.05
max_masks: 3 max_masks: 3
- augmentation_type: frequency_mask - augmentation_type: mask_freq
probability: 0.2 probability: 0.2
max_perc: 0.10 max_perc: 0.10
max_masks: 3 max_masks: 3