From 4d8920099af07f8cd9242c83dbfbc72f73595c84 Mon Sep 17 00:00:00 2001
From: Tao Gong <gongtao950513@gmail.com>
Date: Mon, 9 Aug 2021 16:24:43 +0800
Subject: [PATCH] fix a bug of mot_reid init

---
 mmtrack/models/reid/fc_module.py        |  4 ++--
 mmtrack/models/reid/linear_reid_head.py | 10 +++-------
 2 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/mmtrack/models/reid/fc_module.py b/mmtrack/models/reid/fc_module.py
index f30f97bd0..863e0eae3 100644
--- a/mmtrack/models/reid/fc_module.py
+++ b/mmtrack/models/reid/fc_module.py
@@ -15,7 +15,7 @@ class FcModule(BaseModule):
             Defaults to dict(type='ReLU').
         inplace (bool, optional): Whether inplace the activatation module.
         init_cfg (dict or list[dict], optional): Initialization config dict.
-            Defaults to dict(type='Kaiming', layer='fc').
+            Defaults to dict(type='Kaiming', layer='Linear').
     """
 
     def __init__(self,
@@ -24,7 +24,7 @@ def __init__(self,
                  norm_cfg=None,
                  act_cfg=dict(type='ReLU'),
                  inplace=True,
-                 init_cfg=dict(type='Kaiming', layer='fc')):
+                 init_cfg=dict(type='Kaiming', layer='Linear')):
         super(FcModule, self).__init__(init_cfg)
         assert norm_cfg is None or isinstance(norm_cfg, dict)
         assert act_cfg is None or isinstance(act_cfg, dict)
diff --git a/mmtrack/models/reid/linear_reid_head.py b/mmtrack/models/reid/linear_reid_head.py
index 61d4458fc..446d1d062 100644
--- a/mmtrack/models/reid/linear_reid_head.py
+++ b/mmtrack/models/reid/linear_reid_head.py
@@ -30,8 +30,8 @@ class LinearReIDHead(BaseHead):
             re-identificaiton module.
         topk (int, optional): Calculate topk accuracy. Default to False.
         init_cfg (dict or list[dict], optional): Initialization config dict.
-            Defaults to dict(type='Normal',layer=['fc_out', 'classifier'],
-            mean=0, std=0.01, bias=0).
+            Defaults to dict(type='Normal',layer='Linear', mean=0, std=0.01,
+            bias=0).
     """
 
     def __init__(self,
@@ -46,11 +46,7 @@ def __init__(self,
                  loss_pairwise=None,
                  topk=(1, ),
                  init_cfg=dict(
-                     type='Normal',
-                     layer=['fc_out', 'classifier'],
-                     mean=0,
-                     std=0.01,
-                     bias=0)):
+                     type='Normal', layer='Linear', mean=0, std=0.01, bias=0)):
         super(LinearReIDHead, self).__init__(init_cfg)
         assert isinstance(topk, (int, tuple))
         if isinstance(topk, int):