From f39e4ade70c072133e84ea9a4162eb86f6c09c05 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 25 Nov 2022 21:16:31 +0000 Subject: [PATCH] LayerConvNextGamma: fix config serialisation bug .....this is unlikely to be the problem as this bug is in an unused code path. --- aimodel/src/lib/ai/components/LayerConvNeXtGamma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py b/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py index 2396f86..9f29564 100644 --- a/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py +++ b/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py @@ -15,5 +15,5 @@ class LayerConvNeXtGamma(tf.keras.layers.Layer): def get_config(self): config = super(LayerConvNeXtGamma, self).get_config() - config.update({ "const": self.const.numpy(), "dim": self.dim }) + config.update({ "const_val": self.const.numpy(), "dim": self.dim }) return config