Removed softmax from solveig_model.py

hzavadil98 · web-flow · commit fae506d319db · 2025-02-05T10:37:58.000+01:00
diff --git a/utils/models/solveig_model.py b/utils/models/solveig_model.py
@@ -58,7 +58,6 @@ def forward(self, x):
         x = torch.flatten(x, 1)
 
         x = self.fc1(x)
-        x = nn.Softmax(x)
 
         return x
 
