ultralytics 8.3.39 fix classification validation loss scaling (#17851)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
29826241a0
commit
e60992214c
3 changed files with 3 additions and 3 deletions
|
|
@ -296,7 +296,7 @@ class Classify(nn.Module):
|
|||
if isinstance(x, list):
|
||||
x = torch.cat(x, 1)
|
||||
x = self.linear(self.drop(self.pool(self.conv(x)).flatten(1)))
|
||||
return x if self.training else x.softmax(1)
|
||||
return x
|
||||
|
||||
|
||||
class WorldDetect(Detect):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue