diff --git a/ppdet/modeling/heads/gfl_head.py b/ppdet/modeling/heads/gfl_head.py
index 66be7708985268ce0274b9736301ed65f279168e..fdecaf6b0916a872bf9e6f243ea62336af4d28ce 100644
--- a/ppdet/modeling/heads/gfl_head.py
+++ b/ppdet/modeling/heads/gfl_head.py
@@ -139,7 +139,7 @@ class DGQP(nn.Layer):
                     axis=2, keepdim=True)], axis=2)
         else:
             stat = prob_topk
-        y = F.relu(self.reg_conv1(stat.reshape([N, 4 * self.reg_topk, H, W])))
+        y = F.relu(self.reg_conv1(stat.reshape([N, 4 * self.total_dim, H, W])))
         y = F.sigmoid(self.reg_conv2(y))
         return y