From c98759277c69dd4c19b3d7e7bb58414b6a667d86 Mon Sep 17 00:00:00 2001 From: klemen1999 Date: Tue, 1 Oct 2024 20:02:36 +0200 Subject: [PATCH] removed unnecessary variable capturing in yamls --- configs/classification_heavy_model.yaml | 4 ++-- configs/classification_light_model.yaml | 4 ++-- configs/detection_heavy_model.yaml | 2 +- configs/detection_light_model.yaml | 2 +- configs/example_export.yaml | 14 ++++++++++++-- configs/example_tuning.yaml | 2 +- configs/keypoint_bbox_heavy_model.yaml | 2 +- configs/keypoint_bbox_light_model.yaml | 2 +- configs/segmentation_heavy_model.yaml | 2 +- configs/segmentation_light_model.yaml | 2 +- 10 files changed, 23 insertions(+), 13 deletions(-) diff --git a/configs/classification_heavy_model.yaml b/configs/classification_heavy_model.yaml index d3e7f098..9feda041 100644 --- 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 index 9230409f..32f7d96b 100644 --- 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 index 82045e1c..a20b64dd 100644 --- 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 index ec1258a0..835e877c 100644 --- 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 index bd7e9793..46367b0c 100644 --- 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 index d85f8cf5..9e63c877 100755 --- 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 index 89ea5df6..81f5778b 100644 --- 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 index 08503257..41eaddd3 100644 --- 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 index b7b76414..e9bc16d6 100644 --- 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 index 07bab77a..cd0fecd3 100644 --- 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