Correct hyperparameters

luxonis · Oct 1, 2024 · 83c2b75 · 83c2b75
1 parent ca147ca
commit 83c2b75
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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