Skip to content

Commit c5c3a4e

Browse files
nkovela1copybara-github
authored andcommitted
Creates non-breaking changes where necessary in preparation for switching all of Keras to new serialization format.
PiperOrigin-RevId: 507864605
1 parent 6dc8101 commit c5c3a4e

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tensorflow_graphics/projects/gan/keras_layers.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -393,14 +393,14 @@ def call(
393393
def get_config(self) -> Dict[str, Any]:
394394
"""Returns the config of the layer."""
395395
config = {
396-
'kernel_size':
397-
self._kernel_size,
398-
'filters':
399-
self._filters,
400-
'kernel_initializer':
401-
tf.keras.utils.serialize_keras_object(self._kernel_initializer),
402-
'bias_initializer':
403-
tf.keras.utils.serialize_keras_object(self._bias_initializer),
396+
'kernel_size': self._kernel_size,
397+
'filters': self._filters,
398+
'kernel_initializer': tf.keras.utils.legacy.serialize_keras_object(
399+
self._kernel_initializer
400+
),
401+
'bias_initializer': tf.keras.utils.legacy.serialize_keras_object(
402+
self._bias_initializer
403+
),
404404
}
405405
base_config = super().get_config()
406406
base_config.update(config)

0 commit comments

Comments
 (0)