From d4535cad1cbd36a760f3dad011ca53c250298835 Mon Sep 17 00:00:00 2001 From: Aryan Jassal <84617406+aryanjassal@users.noreply.github.com> Date: Wed, 17 Jul 2024 16:37:41 +1000 Subject: [PATCH] chore: removed redundant assignment in attention block `nh_kd = nk_kd = self.key_dim ** -0.5` Removed the redundant assignment to: `nh_kd = self.key_dim ** -0.5` --- ultralytics/nn/modules/block.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ultralytics/nn/modules/block.py b/ultralytics/nn/modules/block.py index d11c16e0..963f2699 100644 --- a/ultralytics/nn/modules/block.py +++ b/ultralytics/nn/modules/block.py @@ -776,7 +776,7 @@ class Attention(nn.Module): self.head_dim = dim // num_heads self.key_dim = int(self.head_dim * attn_ratio) self.scale = self.key_dim ** -0.5 - nh_kd = nh_kd = self.key_dim * num_heads + nh_kd = self.key_dim * num_heads h = dim + nh_kd * 2 self.qkv = Conv(dim, h, 1, act=False) self.proj = Conv(dim, dim, 1, act=False) @@ -824,4 +824,4 @@ class SCDown(nn.Module): self.cv2 = Conv(c2, c2, k=k, s=s, g=c2, act=False) def forward(self, x): - return self.cv2(self.cv1(x)) \ No newline at end of file + return self.cv2(self.cv1(x))