first commit
This commit is contained in:
36
rtdetr_pytorch/configs/rtdetr/include/optimizer.yml
Normal file
36
rtdetr_pytorch/configs/rtdetr/include/optimizer.yml
Normal file
@@ -0,0 +1,36 @@
|
||||
|
||||
use_ema: True
|
||||
ema:
|
||||
type: ModelEMA
|
||||
decay: 0.9999
|
||||
warmups: 2000
|
||||
|
||||
|
||||
find_unused_parameters: True
|
||||
|
||||
epoches: 72
|
||||
clip_max_norm: 0.1
|
||||
|
||||
optimizer:
|
||||
type: AdamW
|
||||
params:
|
||||
-
|
||||
params: 'backbone'
|
||||
lr: 0.00001
|
||||
-
|
||||
params: '^(?=.*encoder(?=.*bias|.*norm.*weight)).*$'
|
||||
weight_decay: 0.
|
||||
-
|
||||
params: '^(?=.*decoder(?=.*bias|.*norm.*weight)).*$'
|
||||
weight_decay: 0.
|
||||
|
||||
lr: 0.0001
|
||||
betas: [0.9, 0.999]
|
||||
weight_decay: 0.0001
|
||||
|
||||
|
||||
lr_scheduler:
|
||||
type: MultiStepLR
|
||||
milestones: [1000]
|
||||
gamma: 0.1
|
||||
|
||||
Reference in New Issue
Block a user