gnn with weights

2023-02-25 12:33:18 +00:00
parent 5062ed95d8
commit d8337bfcfe
4 changed files with 17 additions and 13 deletions
@@ -258,7 +258,8 @@ class ConvNeXtReverse(tf.keras.Model):
        if not learn_weights:
            self.last = Dense(3, activation='tanh')
        else:
-            self.last = Activation('softmax')
+            # self.last = Activation('softmax')
+            self.last = lambda x, training: x
            # self.last = Dense(24, activation='softmax')