train encoder on raw curvature values

convnext/models/convnext_tf.py

@@ -338,7 +338,7 @@ class ConvNeXtReverseLabels(tf.keras.Model):
     """
 
     def __init__(self, 
-                 depths=[3, 3, 9, 3], dims=[96, 192, 384, 768],
+                 depths=[3, 3, 9, 3], dims=[96, 192, 384, 768], kernel_size=7,
                  drop_path_rate=0., layer_scale_init_value=1e-6, head_init_scale=1.,
                  nr_of_classes=7):
         super().__init__()
@@ -360,14 +360,14 @@ class ConvNeXtReverseLabels(tf.keras.Model):
         cur = 0
         for i in range(len(depths)):
             stage = tf.keras.Sequential(
-                [Block(dim=dims[i], drop_path=dp_rates[cur + j],
+                [Block(dim=dims[i], drop_path=dp_rates[cur + j], kernel_size=kernel_size,
                        layer_scale_init_value=layer_scale_init_value, prefix=f'block{i}') for j in range(depths[i])]
             )
             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(1, activation='sigmoid')
 
     def call(self, x, training):
         for i in reversed(range(len(self.downsample_layers))):