From f8e21c8fc03a10900eaba99fbf0e4f3790eee0b8 Mon Sep 17 00:00:00 2001
From: Wenyu <wenyu.lyu@gmail.com>
Date: Wed, 28 Dec 2022 14:46:09 +0800
Subject: [PATCH] update vitdet yolo optim (#7567)

---
 configs/vitdet/_base_/optimizer_base_30e.yml  |  0
 configs/vitdet/_base_/optimizer_base_36e.yml  | 20 +++++++++++++++++++
 ... ppyoloe_vit_base_csppan_cae_36e_coco.yml} |  4 ++--
 3 files changed, 22 insertions(+), 2 deletions(-)
 delete mode 100644 configs/vitdet/_base_/optimizer_base_30e.yml
 create mode 100644 configs/vitdet/_base_/optimizer_base_36e.yml
 rename configs/vitdet/{ppyoloe_vit_base_csppan_cae_30e_coco.yml => ppyoloe_vit_base_csppan_cae_36e_coco.yml} (93%)

diff --git a/configs/vitdet/_base_/optimizer_base_30e.yml b/configs/vitdet/_base_/optimizer_base_30e.yml
deleted file mode 100644
index e69de29bb..000000000
diff --git a/configs/vitdet/_base_/optimizer_base_36e.yml b/configs/vitdet/_base_/optimizer_base_36e.yml
new file mode 100644
index 000000000..83b8708d0
--- /dev/null
+++ b/configs/vitdet/_base_/optimizer_base_36e.yml
@@ -0,0 +1,20 @@
+
+epoch: 36
+
+LearningRate:
+  base_lr: 0.0001
+  schedulers:
+    - !CosineDecay
+      max_epochs: 36
+      min_lr_ratio: 0.1 # 0.1
+    - !LinearWarmup
+      start_factor: 0.001
+      epochs: 1
+
+
+OptimizerBuilder:
+  clip_grad_by_norm: 0.1
+  regularizer: false
+  optimizer:
+    type: AdamW
+    weight_decay: 0.0001
diff --git a/configs/vitdet/ppyoloe_vit_base_csppan_cae_30e_coco.yml b/configs/vitdet/ppyoloe_vit_base_csppan_cae_36e_coco.yml
similarity index 93%
rename from configs/vitdet/ppyoloe_vit_base_csppan_cae_30e_coco.yml
rename to configs/vitdet/ppyoloe_vit_base_csppan_cae_36e_coco.yml
index d446d5cc9..556f4b49d 100644
--- a/configs/vitdet/ppyoloe_vit_base_csppan_cae_30e_coco.yml
+++ b/configs/vitdet/ppyoloe_vit_base_csppan_cae_36e_coco.yml
@@ -3,10 +3,10 @@ _BASE_: [
   '../datasets/coco_detection.yml',
   '../runtime.yml',
   './_base_/ppyoloe_reader.yml',
-  './_base_/optimizer_base_30e.yml'
+  './_base_/optimizer_base_36e.yml'
 ]
 
-weights: output/ppyoloe_vit_base_csppan_cae_30e_coco/model_final
+weights: output/ppyoloe_vit_base_csppan_cae_36e_coco/model_final
 
 
 snapshot_epoch: 2
-- 
GitLab