From 3d614d105b3c899afb0231995e485b4b4b4740bc Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 31 Aug 2022 18:06:59 +0100 Subject: [PATCH] channels first --- aimodel/src/lib/ai/components/LayerContrastiveEncoder.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py index 654c9d9..9c307be 100644 --- a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py +++ b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py @@ -31,9 +31,10 @@ class LayerContrastiveEncoder(tf.keras.layers.Layer): """ self.encoder = ResNet50V2( 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, - pooling=None + pooling=None, + data_format="channels_first" ) """Small sequential stack of layers that control the size of the outputted feature dimension. """