ultralytics 8.1.29 improved disk space checking on correct path (#8977)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
This commit is contained in:
parent
b44e184041
commit
ab712288f7
4 changed files with 8 additions and 7 deletions
|
|
@ -57,7 +57,7 @@ class DFL(nn.Module):
|
|||
|
||||
def forward(self, x):
|
||||
"""Applies a transformer layer on input tensor 'x' and returns a tensor."""
|
||||
b, c, a = x.shape # batch, channels, anchors
|
||||
b, _, a = x.shape # batch, channels, anchors
|
||||
return self.conv(x.view(b, 4, self.c1, a).transpose(2, 1).softmax(1)).view(b, 4, a)
|
||||
# return self.conv(x.view(b, self.c1, 4, a).softmax(1)).view(b, 4, a)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue