_base_ = ['../mnist_grid/yolov3_d53_fp16_mstrain-608_273e_mnist.py',]

n_freq = 3
model = dict(
    bbox_head=dict(
        type='SpatialRelationYOLOV3Head',
        num_frequencies=n_freq,
    )
)

find_unused_parameters = True
optimizer = dict(_delete_=True,
    type='Adam', lr=1e-5, weight_decay=0.0001)#, momentum=0.9
optimizer_config = dict(grad_clip=None)
lr_config = dict(
    policy='step',
    warmup='linear',
    warmup_iters=200,
    warmup_ratio=0.001,
    step=[6, 10])
runner = dict(type='EpochBasedRunner', max_epochs=12)

load_from = '/data/vision/polina/users/clintonw/code/mmdet/checkpoints/yolov3_d53_fp16_mstrain-608_273e_coco_20210517_213542-4bc34944.pth'
