Correct class end module

golmschenk · Jul 31, 2024 · 7516ed9 · 7516ed9
1 parent dfb73df
commit 7516ed9
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/qusi/internal/hadryss_model.py b/src/qusi/internal/hadryss_model.py
@@ -268,11 +268,9 @@ def __init__(self, number_of_classes: int):
         super().__init__()
         self.number_of_classes: int = number_of_classes
         self.prediction_layer = Conv1d(in_channels=20, out_channels=self.number_of_classes, kernel_size=1)
-        self.soft_max = Softmax(dim=1)
 
     def forward(self, x: Tensor) -> Tensor:
         x = self.prediction_layer(x)
-        x = self.soft_max(x)
         x = torch.reshape(x, (-1, self.number_of_classes))
         return x