baconkcode

change your configs like this model = dict( bbox_head=dict( head_module=dict(num_classes=num_classes)), train_cfg=dict(assigner=dict( type='BatchTaskAlignedAssigner', **num_classes=1,#your num_classes** use_ciou=True, topk=10, alpha=0.5, beta=6.0, eps=1e-09)) )