wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-17 11:20:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix serialization

Browse Source
This commit is contained in:
Michael Oliver
2017-02-22 11:33:43 -08:00
parent 82b92142cf
commit ac44e1a19c
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/layers/core.py
+4
View File
@@ -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})