From 48d8a4d39ce96d357b4761283d855dbbf19111d8 Mon Sep 17 00:00:00 2001
From: Feng Ni <nemonameless@qq.com>
Date: Wed, 22 Sep 2021 10:42:25 +0800
Subject: [PATCH] [cherry-pick][MOT][MOT] fix fairmot centernet_head bias init 
 (#4207)

* fix operators typo

* fix centernet_head bias init
---
 ppdet/modeling/heads/centernet_head.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppdet/modeling/heads/centernet_head.py b/ppdet/modeling/heads/centernet_head.py
index aa3fbe2c3..49c7b0df1 100755
--- a/ppdet/modeling/heads/centernet_head.py
+++ b/ppdet/modeling/heads/centernet_head.py
@@ -98,7 +98,8 @@ class CenterNetHead(nn.Layer):
                 stride=1,
                 padding=0,
                 bias=True))
-        self.heatmap[2].conv.bias[:] = -2.19
+        with paddle.no_grad():
+            self.heatmap[2].conv.bias[:] = -2.19
         self.size = nn.Sequential(
             ConvLayer(
                 in_channels, head_planes, kernel_size=3, padding=1, bias=True),
-- 
GitLab