Removed softmax from johan_model.py

Author: hzavadil98

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