From a8015e4b320b2425c4f89067b0fab7395dddebcd Mon Sep 17 00:00:00 2001
From: FlyingQianMM <245467267@qq.com>
Date: Mon, 21 Jun 2021 11:40:05 +0800
Subject: [PATCH] fix train diff for fairmot (#3436)

---
 ppdet/data/transform/operators.py | 6 +++---
 ppdet/modeling/layers.py          | 2 ++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index 54f918c9b..bd51db369 100644
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -2015,12 +2015,12 @@ class Rbox2Poly(BaseOperator):
 
 @register_op
 class AugmentHSV(BaseOperator):
-    def __init__(self, fraction=0.50, is_bgr=False):
+    def __init__(self, fraction=0.50, is_bgr=True):
         """ 
         Augment the SV channel of image data.
         Args:
-            fraction (float): the fraction for augment 
-            is_bgr (bool): whether the image is BGR mode
+            fraction (float): the fraction for augment. Default: 0.5.
+            is_bgr (bool): whether the image is BGR mode. Default: True.
         """
         super(AugmentHSV, self).__init__()
         self.fraction = fraction
diff --git a/ppdet/modeling/layers.py b/ppdet/modeling/layers.py
index a10d53aeb..919c3d353 100644
--- a/ppdet/modeling/layers.py
+++ b/ppdet/modeling/layers.py
@@ -163,6 +163,8 @@ class ConvNormLayer(nn.Layer):
                 bias_attr=True,
                 lr_scale=dcn_lr_scale,
                 regularizer=dcn_regularizer,
+                dcn_bias_regularizer=dcn_regularizer,
+                dcn_bias_lr_scale=dcn_lr_scale,
                 skip_quant=skip_quant)
 
         norm_lr = 0. if freeze_norm else 1.
-- 
GitLab