调整学习率和训练轮数，更新学习率调度器的总迭代次数，优化优化器配置

2024-10-16 11:11:10 +08:00 · 2024-10-16 11:11:10 +08:00 · e935cf6f85
commit e935cf6f85
parent 20537fe10f
3 changed files with 4 additions and 4 deletions
--- a/config/lr_scheduler.yaml
+++ b/config/lr_scheduler.yaml
@ -4,8 +4,8 @@ scheduler_type: LINEAR_WARMUP_THEN_POLY_SCHEDULER
 kwargs: |
  {
    "warmup_iters": 5808, 
-    "total_iters": 580800, 
+    "total_iters": 290400, 
    "warmup_ratio": 0.000001, 
-    "min_lr": 0.00000, 
+    "min_lr": 0., 
    "power": 1.
  }
--- a/config/optimizer.yaml
+++ b/config/optimizer.yaml
@ -3,7 +3,7 @@ base_config:
  optim_type: AdamW
  kwargs: |
    {
-      "lr": 0.006,
+      "lr": 0.0006,
      "weight_decay": 0.01,
      "betas": (0.9, 0.999)
    }
--- a/config/train.yaml
+++ b/config/train.yaml
@ -3,7 +3,7 @@ batch_size: 16
 image_height: 200
 image_width: 200
 workers: 0
-epochs: 200
+epochs: 100

 # 每一类的占比权重，如果要让每一类的占比权重相同，为1.0即可
 weight: