File size: 972 Bytes
f86c59c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | __include__: [
'../../dataset/custom_detection.yml',
'../../runtime.yml',
'../include/dataloader.yml',
'../include/optimizer.yml',
'../include/dfine_hgnetv2.yml',
]
output_dir: ./output/dfine_hgnetv2_x_custom
DFINE:
backbone: HGNetv2
HGNetv2:
name: 'B5'
return_idx: [1, 2, 3]
freeze_stem_only: True
freeze_at: 0
freeze_norm: True
HybridEncoder:
hidden_dim: 384
dim_feedforward: 2048
DFINETransformer:
feat_channels: [384, 384, 384]
reg_scale: 8
optimizer:
type: AdamW
params:
-
params: '^(?=.*backbone)(?!.*norm|bn).*$'
lr: 0.0000025
-
params: '^(?=.*(?:encoder|decoder))(?=.*(?:norm|bn)).*$'
weight_decay: 0.
lr: 0.00025
betas: [0.9, 0.999]
weight_decay: 0.000125
# Increase to search for the optimal ema
epochs: 35 # 30 + 5
train_dataloader:
dataset:
transforms:
policy:
epoch: 30
collate_fn:
stop_epoch: 30
ema_restart_decay: 0.9998
base_size_repeat: 3
|