From 83c2b759cb855a4cc1c7a03edb75d3a8792e7044 Mon Sep 17 00:00:00 2001 From: Jernej Sabadin Date: Tue, 1 Oct 2024 13:01:50 +0200 Subject: [PATCH] Correct hyperparameters --- configs/detection_light_model.yaml | 2 +- configs/segmentation_light_model.yaml | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/configs/detection_light_model.yaml b/configs/detection_light_model.yaml index ad400f84..88d1184f 100644 --- a/configs/detection_light_model.yaml +++ b/configs/detection_light_model.yaml @@ -20,7 +20,7 @@ trainer: batch_size: 8 epochs: &epochs 200 - n_workers: 8 + n_workers: 4 validation_interval: 10 n_log_images: 8 diff --git a/configs/segmentation_light_model.yaml b/configs/segmentation_light_model.yaml index 9147c509..4454260f 100644 --- a/configs/segmentation_light_model.yaml +++ b/configs/segmentation_light_model.yaml @@ -32,12 +32,13 @@ trainer: optimizer: name: SGD params: - lr: 0.01 + lr: 0.02 momentum: 0.9 weight_decay: 0.0005 nesterov: true - + scheduler: name: CosineAnnealingLR params: T_max: *epochs + eta_min: 0.0002 \ No newline at end of file