diff --git a/mmtrack/models/backbones/sot_resnet.py b/mmtrack/models/backbones/sot_resnet.py index a77d8387b..6a8cad72f 100644 --- a/mmtrack/models/backbones/sot_resnet.py +++ b/mmtrack/models/backbones/sot_resnet.py @@ -18,7 +18,8 @@ def __init__(self, conv_cfg=None, norm_cfg=dict(type='BN'), dcn=None, - plugins=None): + plugins=None, + init_cfg=None): """Bottleneck block for ResNet. If style is "pytorch", the stride-two layer is the 3x3 conv layer, if @@ -35,7 +36,8 @@ def __init__(self, conv_cfg=None, norm_cfg=dict(type='BN'), dcn=None, - plugins=None) + plugins=None, + init_cfg=init_cfg) assert style in ['pytorch', 'caffe'] assert dcn is None or isinstance(dcn, dict) assert plugins is None or isinstance(plugins, list)