Update config to remove optimiser level

This commit is contained in:
mbsantiago 2025-06-26 18:42:53 -06:00
parent 15de168a20
commit d67ae9be05

View File

@ -101,6 +101,8 @@ model:
train: train:
batch_size: 8 batch_size: 8
learning_rate: 0.001
t_max: 100
loss: loss:
detection: detection:
weight: 1.0 weight: 1.0
@ -114,9 +116,6 @@ train:
alpha: 2 alpha: 2
size: size:
weight: 0.1 weight: 0.1
optimizer:
learning_rate: 0.001
t_max: 100
logger: logger:
logger_type: dvclive logger_type: dvclive
augmentations: augmentations: