Config Anpassungen

2020-04-15 17:15:51 +02:00
parent 427a6463cb
commit 76c0e6aa05
4 changed files with 14 additions and 3 deletions
--- a/models/binary_classifier.py
+++ b/models/binary_classifier.py
@@ -31,11 +31,14 @@ class BinaryClassifier(LightningBaseModule):

    def __init__(self, hparams):
        super(BinaryClassifier, self).__init__(hparams)
+
        self.criterion = nn.BCELoss()

        # Additional parameters
        self.in_shape = ()

+        #
+
        # Model Modules
        self.conv_1 = ConvModule(self.in_shape, 32, 5, )
        self.conv_2 = ConvModule(64)