first commit
This commit is contained in:
40
rtdetr_paddle/configs/rtdetr/rtdetr_hgnetv2_x_6x_coco.yml
Normal file
40
rtdetr_paddle/configs/rtdetr/rtdetr_hgnetv2_x_6x_coco.yml
Normal file
@@ -0,0 +1,40 @@
|
||||
_BASE_: [
|
||||
'../datasets/coco_detection.yml',
|
||||
'../runtime.yml',
|
||||
'_base_/optimizer_6x.yml',
|
||||
'_base_/rtdetr_r50vd.yml',
|
||||
'_base_/rtdetr_reader.yml',
|
||||
]
|
||||
|
||||
weights: output/rtdetr_hgnetv2_l_6x_coco/model_final
|
||||
pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/PPHGNetV2_X_ssld_pretrained.pdparams
|
||||
find_unused_parameters: True
|
||||
log_iter: 200
|
||||
|
||||
|
||||
|
||||
DETR:
|
||||
backbone: PPHGNetV2
|
||||
|
||||
|
||||
PPHGNetV2:
|
||||
arch: 'X'
|
||||
return_idx: [1, 2, 3]
|
||||
freeze_stem_only: True
|
||||
freeze_at: 0
|
||||
freeze_norm: True
|
||||
lr_mult_list: [0., 0.01, 0.01, 0.01, 0.01]
|
||||
|
||||
|
||||
HybridEncoder:
|
||||
hidden_dim: 384
|
||||
use_encoder_idx: [2]
|
||||
num_encoder_layers: 1
|
||||
encoder_layer:
|
||||
name: TransformerLayer
|
||||
d_model: 384
|
||||
nhead: 8
|
||||
dim_feedforward: 2048
|
||||
dropout: 0.
|
||||
activation: 'gelu'
|
||||
expansion: 1.0
|
||||
Reference in New Issue
Block a user