mirror of https://github.com/open-mmlab/mmyolo.git
22 lines
839 B
Python
22 lines
839 B
Python
_base_ = './yolov6_s_syncbn_fast_8xb32-300e_coco.py'
|
|
|
|
# ======================= Possible modified parameters =======================
|
|
# -----model related-----
|
|
# The scaling factor that controls the depth of the network structure
|
|
deepen_factor = 0.33
|
|
# The scaling factor that controls the width of the network structure
|
|
widen_factor = 0.25
|
|
|
|
# -----train val related-----
|
|
lr_factor = 0.02 # Learning rate scaling factor
|
|
|
|
# ============================== Unmodified in most cases ===================
|
|
model = dict(
|
|
backbone=dict(deepen_factor=deepen_factor, widen_factor=widen_factor),
|
|
neck=dict(deepen_factor=deepen_factor, widen_factor=widen_factor),
|
|
bbox_head=dict(
|
|
head_module=dict(widen_factor=widen_factor),
|
|
loss_bbox=dict(iou_mode='siou')))
|
|
|
|
default_hooks = dict(param_scheduler=dict(lr_factor=lr_factor))
|