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