lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:keras-team/keras-core

Browse Source
This commit is contained in:
Francois Chollet 2023-04-22 09:08:31 -07:00
parent 54ff92a5d7
commit 53a858cb7f
3 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
keras_core/activations/__init__.py
View File

@ -15,3 +15,7 @@ def get(identifier):
if identifier == "relu":
return relu
return identifier
def serialize(activation):
return activation.__name__

5
keras_core/layers/core/dense.py
View File

@ -61,10 +61,15 @@ class Dense(Layer):
# TODO
config = {
"units": self.units,
"activation": activations.serialize(self.activation),
"use_bias": self.use_bias,
"kernel_initializer": initializers.serialize(
self.kernel_initializer
),
"bias_initializer": initializers.serialize(self.bias_initializer),
"kernel_regularizer": regularizers.serialize(self.kernel_regularizer),
"bias_regularizer": regularizers.serialize(self.bias_regularizer),
"kernel_constraint": constraints.serialize(self.kernel_constraint),
"bias_constraint": constraints.serialize(self.bias_constraint),
}
return {**base_config, **config}

4
keras_core/saving/serialization_lib.py
View File

@ -146,7 +146,7 @@ def serialize_keras_object(obj):
"class_name": "__tensor__",
"config": {
"value": np.array(obj).tolist(),
"dtype": str(obj.dtype),
"dtype": backend.standardize_dtype(obj.dtype),
},
}
if type(obj).__module__ == np.__name__:
@ -155,7 +155,7 @@ def serialize_keras_object(obj):
"class_name": "__numpy__",
"config": {
"value": obj.tolist(),
"dtype": backend.standardize_dytpe(obj.dtype),
"dtype": backend.standardize_dtype(obj.dtype),
},
}
else: