sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-07-02 11:24:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

channels first

Browse source
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
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
aimodel/src/lib/ai/components/LayerContrastiveEncoder.py
View file

@ -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.
"""