From 3cf99587e4bb0b86fd2466235638077443823f58 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 27 Sep 2022 15:53:52 +0100 Subject: [PATCH] Contraster: add gamma layer to load_model --- aimodel/src/lib/ai/RainfallWaterContraster.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index 1d1998d..3196dc7 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -72,6 +72,7 @@ class RainfallWaterContraster(object): self.model_predict = tf.keras.models.load_model(filepath_checkpoint, custom_objects={ "LayerContrastiveEncoder": LayerContrastiveEncoder, + "LayerConvNeXtGamma": LayerConvNeXtGamma, "LayerCheeseMultipleOut": LayerCheeseMultipleOut })