Removed softmax from johan_model.py

SFI-Visual-Intelligence · Feb 5, 2025 · 2c4db88 · 2c4db88
1 parent d767754
commit 2c4db88
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/utils/models/johan_model.py b/utils/models/johan_model.py
@@ -43,14 +43,12 @@ def __init__(self, image_shape, num_classes):
         self.fc2 = nn.Linear(77, 77)
         self.fc3 = nn.Linear(77, 77)
         self.fc4 = nn.Linear(77, num_classes)
-        self.softmax = nn.Softmax(dim=1)
         self.relu = nn.ReLU()
 
     def forward(self, x):
         for layer in [self.fc1, self.fc2, self.fc3, self.fc4]:
             x = layer(x)
             x = self.relu(x)
-        x = self.softmax(x)
         return x