OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
489 B
14 lines
489 B
_base_ = [ |
|
'../_base_/models/retinanet_r50_fpn.py', '../_base_/datasets/voc0712.py', |
|
'../_base_/default_runtime.py' |
|
] |
|
model = dict(bbox_head=dict(num_classes=20)) |
|
# optimizer |
|
optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) |
|
optimizer_config = dict(grad_clip=None) |
|
# learning policy |
|
# actual epoch = 3 * 3 = 9 |
|
lr_config = dict(policy='step', step=[3]) |
|
# runtime settings |
|
runner = dict( |
|
type='EpochBasedRunner', max_epochs=4) # actual epoch = 4 * 3 = 12
|
|
|