48 lines
821 B
YAML
48 lines
821 B
YAML
model:
|
|
decoder:
|
|
backbone:
|
|
type: SECOND
|
|
in_channels: 256
|
|
out_channels: [128, 256]
|
|
layer_nums: [5, 5]
|
|
layer_strides: [1, 2]
|
|
norm_cfg:
|
|
type: BN
|
|
eps: 1.0e-3
|
|
momentum: 0.01
|
|
conv_cfg:
|
|
type: Conv2d
|
|
bias: false
|
|
neck:
|
|
type: SECONDFPN
|
|
in_channels: [128, 256]
|
|
out_channels: [256, 256]
|
|
upsample_strides: [1, 2]
|
|
norm_cfg:
|
|
type: BN
|
|
eps: 1.0e-3
|
|
momentum: 0.01
|
|
upsample_cfg:
|
|
type: deconv
|
|
bias: false
|
|
use_conv_for_no_stride: true
|
|
heads:
|
|
object:
|
|
in_channels: 512
|
|
|
|
optimizer:
|
|
type: AdamW
|
|
lr: 1.0e-4
|
|
weight_decay: 0.01
|
|
|
|
optimizer_config:
|
|
grad_clip:
|
|
max_norm: 35
|
|
norm_type: 2
|
|
|
|
lr_config:
|
|
policy: cyclic
|
|
|
|
momentum_config:
|
|
policy: cyclic
|