removed unnecessary variable capturing in yamls

luxonis · Oct 1, 2024 · c987592 · c987592
1 parent 263fe43
commit c987592
Show file tree

Hide file tree

Showing 10 changed files with 23 additions and 13 deletions.
diff --git a/configs/classification_heavy_model.yaml b/configs/classification_heavy_model.yaml
@@ -13,13 +13,13 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True
 
   batch_size: 8
-  epochs: &epochs 200
+  epochs: 200
   n_workers: 4
   validation_interval: 10
   n_log_images: 8

diff --git a/configs/classification_light_model.yaml b/configs/classification_light_model.yaml
@@ -13,13 +13,13 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True
 
   batch_size: 8
-  epochs: &epochs 200
+  epochs: 200
   n_workers: 4
   validation_interval: 10
   n_log_images: 8

diff --git a/configs/detection_heavy_model.yaml b/configs/detection_heavy_model.yaml
@@ -13,7 +13,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True

diff --git a/configs/detection_light_model.yaml b/configs/detection_light_model.yaml
@@ -13,7 +13,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True

diff --git a/configs/example_export.yaml b/configs/example_export.yaml
@@ -14,7 +14,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True
@@ -27,9 +27,19 @@ trainer:
 
   optimizer:
     name: SGD
+    params:
+      lr: 0.02             
+      momentum: 0.937    
+      weight_decay: 0.0005   
+      dampening: 0.0       
+      nesterov: true     
 
   scheduler:
-    name: ConstantLR
+    name: CosineAnnealingLR
+    params:
+      T_max: *epochs   
+      eta_min: 0.0002 
+      last_epoch: -1
 
 exporter:
   output_names: [output1_yolov6r2, output2_yolov6r2, output3_yolov6r2]

diff --git a/configs/example_tuning.yaml b/configs/example_tuning.yaml
@@ -14,7 +14,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True

diff --git a/configs/keypoint_bbox_heavy_model.yaml b/configs/keypoint_bbox_heavy_model.yaml
@@ -13,7 +13,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True

diff --git a/configs/keypoint_bbox_light_model.yaml b/configs/keypoint_bbox_light_model.yaml
@@ -13,7 +13,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True

diff --git a/configs/segmentation_heavy_model.yaml b/configs/segmentation_heavy_model.yaml
@@ -13,7 +13,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True

diff --git a/configs/segmentation_light_model.yaml b/configs/segmentation_light_model.yaml
@@ -13,7 +13,7 @@ loader:
 
 trainer:
   preprocessing:
-    train_image_size: [&height 384, &width 512]
+    train_image_size: [384, 512]
     keep_aspect_ratio: True
     normalize:
       active: True