train curv encoders with 10p and radius 10,25

2023-01-28 15:12:42 +00:00
parent af336370d4
commit 930668611f
7 changed files with 149 additions and 44 deletions
@@ -366,8 +366,8 @@ class ConvNeXtReverseLabels(tf.keras.Model):
            self.stages.append(stage)
            cur += depths[i]

-        # self.last = Dense(nr_of_classes, activation='softmax')
-        self.last = Dense(nr_of_classes - 1)
+        self.last = Dense(nr_of_classes, activation='softmax')
+        # self.last = Dense(nr_of_classes - 1)

    def call(self, x, training):
        for i in reversed(range(len(self.downsample_layers))):