From 81544c6d71f4c5682e252e14a68f05e994ed7151 Mon Sep 17 00:00:00 2001 From: Aryan Jassal <84617406+aryanjassal@users.noreply.github.com> Date: Thu, 18 Jul 2024 01:27:07 +1000 Subject: [PATCH] Remove redundant assignment (#14498) Co-authored-by: Glenn Jocher --- ultralytics/nn/modules/block.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/nn/modules/block.py b/ultralytics/nn/modules/block.py index 9d08dd7e..9b54353e 100644 --- a/ultralytics/nn/modules/block.py +++ b/ultralytics/nn/modules/block.py @@ -855,7 +855,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)