KeyError: 'SOLOv2 is not in the models registry'

[yyyy1221]

Traceback (most recent call last):
  File "onnx_exporter_solov2.py", line 222, in <module>
    convert2onnx(args, dummy_input)
  File "onnx_exporter_solov2.py", line 87, in convert2onnx
    model = init_detector(args.config, args.checkpoint, device='cuda:0')
  File "/usr/local/lib/python3.6/dist-packages/mmdet/apis/inference.py", line 39, in init_detector
    model = build_detector(config.model, test_cfg=config.get('test_cfg'))
  File "/usr/local/lib/python3.6/dist-packages/mmdet/models/builder.py", line 58, in build_detector
    cfg, default_args=dict(train_cfg=train_cfg, test_cfg=test_cfg))
  File "/usr/local/lib/python3.6/dist-packages/mmcv/utils/registry.py", line 210, in build
    return self.build_func(*args, **kwargs, registry=self)
  File "/usr/local/lib/python3.6/dist-packages/mmcv/cnn/builder.py", line 26, in build_model_from_cfg
    return build_from_cfg(cfg, registry, default_args)
  File "/usr/local/lib/python3.6/dist-packages/mmcv/utils/registry.py", line 44, in build_from_cfg
    f'{obj_type} is not in the {registry.name} registry')
KeyError: 'SOLOv2 is not in the models registry'

I'm running on docker, but I keep getting this error.
Any help would be appreciated.