channels first

This commit is contained in:
Starbeamrainbowlabs 2022-08-31 18:06:59 +01:00
parent 654eefd9ca
commit 3d614d105b
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -31,9 +31,10 @@ class LayerContrastiveEncoder(tf.keras.layers.Layer):
""" """
self.encoder = ResNet50V2( self.encoder = ResNet50V2(
include_top=False, include_top=False,
input_shape=(self.param_input_width, self.param_input_height, self.param_channels), input_shape=(self.param_channels, self.param_input_width, self.param_input_height),
weights=None, weights=None,
pooling=None pooling=None,
data_format="channels_first"
) )
"""Small sequential stack of layers that control the size of the outputted feature dimension. """Small sequential stack of layers that control the size of the outputted feature dimension.
""" """