From 6862c9850a4a7577ed7227f850c83153f328a066 Mon Sep 17 00:00:00 2001
From: SeeFun <seefun@outlook.com>
Date: Wed, 27 Dec 2023 16:49:37 +0800
Subject: [PATCH] fix backward in hgnet

---
 timm/models/hgnet.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/timm/models/hgnet.py b/timm/models/hgnet.py
index 59c7a870..ce2786bf 100644
--- a/timm/models/hgnet.py
+++ b/timm/models/hgnet.py
@@ -23,8 +23,8 @@ class LearnableAffineBlock(nn.Module):
                  scale_value=1.0,
                  bias_value=0.0):
         super().__init__()
-        self.scale = nn.Parameter(torch.tensor([scale_value]))
-        self.bias = nn.Parameter(torch.tensor([bias_value]))
+        self.scale = nn.Parameter(torch.tensor([scale_value]), requires_grad=True)
+        self.bias = nn.Parameter(torch.tensor([bias_value]), requires_grad=True)
 
     def forward(self, x):
         return self.scale * x + self.bias
@@ -262,7 +262,7 @@ class HGBlock(nn.Module):
         x = torch.cat(output, dim=1)
         x = self.aggregation(x)
         if self.residual:
-            x += identity
+            x = x + identity
         return x