From cb3e1d3a232ad82966c452b227e91b9efce740af Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 1 Sep 2022 16:10:50 +0100 Subject: [PATCH] ai: fixup --- aimodel/src/lib/ai/model_rainfallwater_contrastive.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py index dba44a9..f7e18de 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py +++ b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py @@ -37,8 +37,9 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, feature_dim=204 )(input_water) - final = LayerCheeseMultipleOut()([ rainfall, water ]) - weight_temperature = final.weight_temperature + layer_final = LayerCheeseMultipleOut() + final = layer_final([ rainfall, water ]) + weight_temperature = layer_final.weight_temperature model = tf.keras.Model( inputs = [ input_rainfall, input_water ],