lot of things

2022-08-06 12:27:24 +00:00
parent cd498e79f3
commit ee8f1cb68c
28 changed files with 4595 additions and 175 deletions
@@ -91,7 +91,7 @@ class GarmentEmbedding(tf.keras.layers.Layer):

        # TODO: channels_first?

-        self.encoder = ConvNeXt(depths=[3, 3, 27, 3], dims=[96, 192, 256, self.embed_dim], include_top=False)
+        self.encoder = ConvNeXt(depths=[3, 3, 9, 3], dims=[32, 64, 128, self.embed_dim], include_top=False)
        # self.encoder(tf.keras.Input(shape=(*input_shape, 3)))
        self.pooling = tf.keras.layers.GlobalAveragePooling2D()