[Fix] fix a bug in sot resnet backbone

open-mmlab · May 13, 2021 · cc0f9c6 · cc0f9c6
1 parent 56f2383
commit cc0f9c6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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)