From cc0f9c6c3e5fd20ddec764ca2392a16aa61c7fc8 Mon Sep 17 00:00:00 2001 From: ToumaKazusa3 Date: Thu, 13 May 2021 22:10:08 +0800 Subject: [PATCH] [Fix] fix a bug in sot resnet backbone --- mmtrack/models/backbones/sot_resnet.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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)