Skip to content

Commit 48d8a4d

Browse files
authored
[cherry-pick][MOT][MOT] fix fairmot centernet_head bias init (#4207)
* fix operators typo * fix centernet_head bias init
1 parent 40a8249 commit 48d8a4d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ppdet/modeling/heads/centernet_head.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,8 @@ def __init__(self,
9898
stride=1,
9999
padding=0,
100100
bias=True))
101-
self.heatmap[2].conv.bias[:] = -2.19
101+
with paddle.no_grad():
102+
self.heatmap[2].conv.bias[:] = -2.19
102103
self.size = nn.Sequential(
103104
ConvLayer(
104105
in_channels, head_planes, kernel_size=3, padding=1, bias=True),

0 commit comments

Comments
 (0)