new weight trials, 10_3 didn't run

2023-02-26 11:04:45 +00:00
parent d8337bfcfe
commit 03b5eae0d8
3 changed files with 9 additions and 9 deletions
@@ -258,8 +258,8 @@ class ConvNeXtReverse(tf.keras.Model):
        if not learn_weights:
            self.last = Dense(3, activation='tanh')
        else:
-            # self.last = Activation('softmax')
-            self.last = lambda x, training: x
+            self.last = Activation('softmax')
+            # self.last = lambda x, training: x
            # self.last = Dense(24, activation='softmax')