pdch/SegNeXt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
陈培栋 2024-10-16 11:11:10 +08:00
parent 20537fe10f
commit e935cf6f85
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config/lr_scheduler.yaml
View File

@ -4,8 +4,8 @@ scheduler_type: LINEAR_WARMUP_THEN_POLY_SCHEDULER
kwargs: | kwargs: |
{ {
"warmup_iters": 5808, "warmup_iters": 5808,
"total_iters": 580800, "total_iters": 290400,
"warmup_ratio": 0.000001, "warmup_ratio": 0.000001,
"min_lr": 0.00000, "min_lr": 0.,
"power": 1. "power": 1.
} }

2
config/optimizer.yaml
View File

@ -3,7 +3,7 @@ base_config:
optim_type: AdamW optim_type: AdamW
kwargs: | kwargs: |
{ {
"lr": 0.006, "lr": 0.0006,
"weight_decay": 0.01, "weight_decay": 0.01,
"betas": (0.9, 0.999) "betas": (0.9, 0.999)
} }

2
config/train.yaml
View File

@ -3,7 +3,7 @@ batch_size: 16
image_height: 200 image_height: 200
image_width: 200 image_width: 200
workers: 0 workers: 0
epochs: 200 epochs: 100
# 每一类的占比权重，如果要让每一类的占比权重相同，为1.0即可 # 每一类的占比权重，如果要让每一类的占比权重相同，为1.0即可
weight: weight: