From 838ff56a3be0f154b2c33f2b744e8e184c6451b2 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 29 Nov 2022 15:25:11 +0000 Subject: [PATCH] mono: fix loading checkpoint --- aimodel/src/lib/ai/RainfallWaterMono.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/RainfallWaterMono.py b/aimodel/src/lib/ai/RainfallWaterMono.py index dbfed1c..873b3d6 100644 --- a/aimodel/src/lib/ai/RainfallWaterMono.py +++ b/aimodel/src/lib/ai/RainfallWaterMono.py @@ -15,6 +15,7 @@ from .helpers import make_callbacks from .helpers import summarywriter from .components.LayerConvNeXtGamma import LayerConvNeXtGamma from .components.LayerStack2Image import LayerStack2Image +from .components.LossCrossentropy import LossCrossentropy from .helpers.summarywriter import summarywriter class RainfallWaterMono(object): @@ -71,7 +72,8 @@ class RainfallWaterMono(object): self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={ "LayerConvNeXtGamma": LayerConvNeXtGamma, - "LayerStack2Image": LayerStack2Image + "LayerStack2Image": LayerStack2Image, + "LossCrossentropy": LossCrossentropy })