mirror of https://github.com/open-mmlab/mmyolo.git
16 lines
421 B
Python
16 lines
421 B
Python
_base_ = './ppyoloe_s_fast_8xb32-300e_coco.py'
|
|
|
|
deepen_factor = 1.33
|
|
widen_factor = 1.25
|
|
|
|
# TODO: training on ppyoloe need to be implemented.
|
|
train_batch_size_per_gpu = 16
|
|
|
|
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)))
|