From ac44e1a19cb3fccb769dbc22b5cf93a0f29228e3 Mon Sep 17 00:00:00 2001 From: Michael Oliver Date: Wed, 22 Feb 2017 11:33:43 -0800 Subject: [PATCH] fix serialization --- keras_contrib/layers/core.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/keras_contrib/layers/core.py b/keras_contrib/layers/core.py index 4048a90..47de531 100644 --- a/keras_contrib/layers/core.py +++ b/keras_contrib/layers/core.py @@ -21,6 +21,7 @@ from keras.engine import Merge from keras.utils.generic_utils import func_dump from keras.utils.generic_utils import func_load from keras.utils.generic_utils import get_from_module +from keras.utils.generic_utils import get_custom_objects class CosineDense(Layer): @@ -176,3 +177,6 @@ class CosineDense(Layer): 'input_dim': self.input_dim} base_config = super(CosineDense, self).get_config() return dict(list(base_config.items()) + list(config.items())) + + +get_custom_objects().update({"CosineDense": CosineDense})