| _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 | |