From 3c61e2194c7f419627ea5c35b6ff1d403b47b85f Mon Sep 17 00:00:00 2001
From: "alec.tu" <alec.tu@acer.com>
Date: Tue, 29 Aug 2023 11:18:46 +0800
Subject: [PATCH] fix fx tracer to support alpha in RepVGGBlock

---
 mmyolo/models/layers/yolo_bricks.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mmyolo/models/layers/yolo_bricks.py b/mmyolo/models/layers/yolo_bricks.py
index 19175be1a..2ef136228 100644
--- a/mmyolo/models/layers/yolo_bricks.py
+++ b/mmyolo/models/layers/yolo_bricks.py
@@ -183,6 +183,7 @@ def __init__(self,
         else:
             self.se = nn.Identity()
 
+        self.use_alpha=use_alpha
         if use_alpha:
             alpha = torch.ones([
                 1,
@@ -246,7 +247,7 @@ def forward(self, inputs: Tensor) -> Tensor:
             id_out = 0
         else:
             id_out = self.rbr_identity(inputs)
-        if self.alpha:
+        if self.use_alpha:
             return self.nonlinearity(
                 self.se(
                     self.rbr_dense(inputs) +