Removed softmax from christian_model.py

Author: hzavadil98

utils/models/christian_model.py

@@ -58,15 +58,13 @@ def __init__(self, image_shape, num_classes):
         self.cnn2 = CNNBlock(50, 100)
 
         self.fc1 = nn.Linear(100 * 4 * 4, num_classes)
-        self.softmax = nn.Softmax(dim=1)
 
     def forward(self, x):
         x = self.cnn1(x)
         x = self.cnn2(x)
 
         x = x.view(x.size(0), -1)
         x = self.fc1(x)
-        x = self.softmax(x)
 
         return x