From c2fcb3b954cdcd2ce6b87f04650285efdc78c5cf Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 1 Sep 2022 17:06:18 +0100 Subject: [PATCH] =?UTF-8?q?ai:=20channels=5Ffirst=20=E2=86=92=20channels?= =?UTF-8?q?=5Flast?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- aimodel/src/lib/ai/model_rainfallwater_contrastive.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py index f607506..f36fa85 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py +++ b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py @@ -11,7 +11,7 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, batch_size=64, logger.info(shape_water) # Shapes come from what rainfallwrangler sees them as, but we add an extra dimension when reading the .tfrecord file - rainfall_width, rainfall_height, rainfall_channels = shape_rainfall # shape = [width, height, channels] + rainfall_channels, rainfall_width, rainfall_height = shape_rainfall # shape = [channels, width, height] water_width, water_height = shape_water # shape = [width, height] water_channels = 1 # added in dataset → make_dataset → parse_item